1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

SDNQ add ufp aliases

pull/4718/head
Disty0 2026-03-27 21:31:36 +03:00
parent 64400a0cde
commit 6eadf40f33
2 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
modules/sdnq/common.py
View File

@ -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",

1
modules/sdnq/packed_int/__init__.py
View File

@ -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"]