first commit

2026-10-23 15:00:59 +08:00
commit 4627899899
156 changed files with 398139 additions and 0 deletions
--- a/dev/functions/dequant_register.py
+++ b/dev/functions/dequant_register.py
@@ -0,0 +1,20 @@
+from .specs import DequantSpec
+from core.kernels import (dequant_fp8_block_sym, 
+                          dequant_int4_group_sym,
+                          dequant_fp8_channel_sym)
+
+DEQUANT_KERNEL_TABLE = {}
+
+DEQUANT_KERNEL_TABLE[
+    DequantSpec("float", 8, False, "block", True)
+] = dequant_fp8_block_sym
+
+
+DEQUANT_KERNEL_TABLE[
+    DequantSpec("int", 4, True, "group", True)
+] = dequant_int4_group_sym
+
+
+DEQUANT_KERNEL_TABLE[
+    DequantSpec("float", 8, False, "channel", True)
+] = dequant_fp8_channel_sym