diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py index 10351d3ba..f80d902e7 100644 --- a/modules/sdnq/common.py +++ b/modules/sdnq/common.py @@ -239,9 +239,27 @@ dtype_dict["fp5"] = dtype_dict["float5_e2m2fn"] dtype_dict["fp4"] = dtype_dict["float4_e2m1fn"] dtype_dict["fp3"] = dtype_dict["float3_e1m1fn"] dtype_dict["fp2"] = dtype_dict["float2_e1m0fn"] -dtype_dict["fp1"] = dtype_dict["float1_e1m0fnu"] -dtype_dict["bool"] = dtype_dict["uint1"] + +dtype_dict["ufp16"] = dtype_dict["float16_e5m11fnu"] +dtype_dict["ufp15"] = dtype_dict["float15_e5m10fnu"] +dtype_dict["ufp14"] = dtype_dict["float14_e5m9fnu"] +dtype_dict["ufp13"] = dtype_dict["float13_e5m8fnu"] +dtype_dict["ufp12"] = dtype_dict["float12_e5m7fnu"] +dtype_dict["ufp11"] = dtype_dict["float11_e5m6fnu"] +dtype_dict["ufp10"] = dtype_dict["float10_e5m5fnu"] +dtype_dict["ufp9"] = dtype_dict["float9_e4m5fnu"] +dtype_dict["ufp8"] = dtype_dict["float8_e4m4fnu"] +dtype_dict["ufp7"] = dtype_dict["float7_e3m4fnu"] +dtype_dict["ufp6"] = dtype_dict["float6_e3m3fnu"] +dtype_dict["ufp5"] = dtype_dict["float5_e2m3fnu"] +dtype_dict["ufp4"] = dtype_dict["float4_e2m2fnu"] +dtype_dict["ufp3"] = dtype_dict["float3_e1m2fnu"] +dtype_dict["ufp2"] = dtype_dict["float2_e1m1fnu"] +dtype_dict["ufp1"] = dtype_dict["float1_e1m0fnu"] + +dtype_dict["fp1"] = dtype_dict["ufp1"] dtype_dict["int1"] = dtype_dict["uint1"] +dtype_dict["bool"] = dtype_dict["uint1"] torch_dtype_dict = { torch.int32: "int32", diff --git a/modules/sdnq/packed_int/__init__.py b/modules/sdnq/packed_int/__init__.py index ddd6d171d..9fac3314b 100644 --- a/modules/sdnq/packed_int/__init__.py +++ b/modules/sdnq/packed_int/__init__.py @@ -68,6 +68,7 @@ packed_int_function_dict["int5"] = packed_int_function_dict["uint5"] packed_int_function_dict["int4"] = packed_int_function_dict["uint4"] packed_int_function_dict["int3"] = packed_int_function_dict["uint3"] packed_int_function_dict["int2"] = packed_int_function_dict["uint2"] +packed_int_function_dict["int1"] = packed_int_function_dict["uint1"] packed_int_function_dict["bool"] = packed_int_function_dict["uint1"]