first commit

dev/functions/config_builder/kimik2.py

@@ -0,0 +1,38 @@
+# config_builder/qwen3.py
+import re
+
+from .base import QuantConfigBuilder
+from ..specs import ResolvedQuantConfig
+
+
+class KimiK2QuantConfigBuilder(QuantConfigBuilder):
+
+    def __init__(self, qconfig: dict, custom_dequant_config: dict):
+        qc = qconfig
+        self.group_size = qconfig['config_groups']['group_0']['weights']['group_size']
+        self.original_ignore = self.get_ignore(qc)
+        self.custom_ignore = self.get_ignore(custom_dequant_config)
+        self.exact_ignore, self.regex_ignore = self.compile_ignore(self.original_ignore+self.custom_ignore)
+
+    def _ignored(self, weight_name: str) -> bool:
+        if any(weight_name.startswith(ig) for ig in self.exact_ignore):
+            return True
+        if any(re.match(p, weight_name) for p in self.regex_ignore):
+            return True
+        
+        return False
+
+    def resolve(self, weight_name: str):
+        if self._ignored(weight_name):
+            return None
+
+        return ResolvedQuantConfig(
+            dtype="int",
+            num_bits=4,
+            packed=True,
+            strategy="group",
+            group_size=self.group_size,
+            block_structure=None,
+            symmetric=True,
+            dynamic=False,
+        )