init v0.23.0

Signed-off-by: Sun Ruoxi <sunruoxi@4paradigm.com>

tests/ut/quantization/methods/test_registry.py

@@ -0,0 +1,70 @@
+from tests.ut.base import TestBase
+from vllm_ascend.quantization.methods.base import (
+    AscendLinearScheme,
+    AscendMoEScheme,
+)
+from vllm_ascend.quantization.methods.registry import (
+    _SCHEME_REGISTRY,
+    get_scheme_class,
+    register_scheme,
+)
+
+
+class TestRegisterScheme(TestBase):
+    def test_register_scheme(self):
+        @register_scheme("TEST_QUANT_TYPE", "linear")
+        class TestLinearScheme(AscendLinearScheme):
+            def get_weight(self, input_size, output_size, params_dtype):
+                return {}
+
+            def apply(self, layer, x, bias=None, tp_rank=0):
+                return x
+
+        scheme_class = get_scheme_class("TEST_QUANT_TYPE", "linear")
+        self.assertIs(scheme_class, TestLinearScheme)
+
+    def test_register_scheme_duplicate_raises(self):
+        with self.assertRaises(ValueError):
+
+            @register_scheme("W8A8_DYNAMIC", "linear")
+            class Duplicate:
+                pass
+
+
+class TestGetSchemeClass(TestBase):
+    def test_get_existing_scheme_class_existing_linear(self):
+        cls = get_scheme_class("W8A8_DYNAMIC", "linear")
+        self.assertIsNotNone(cls)
+        self.assertTrue(issubclass(cls, AscendLinearScheme))
+
+        cls = get_scheme_class("W8A8_DYNAMIC", "moe")
+        self.assertIsNotNone(cls)
+        self.assertTrue(issubclass(cls, AscendMoEScheme))
+
+        cls = get_scheme_class("FAKQuant", "attention")
+        self.assertIsNotNone(cls)
+
+    def test_get_nonexistent_scheme_class(self):
+        cls = get_scheme_class("NONEXISTENT", "linear")
+        self.assertIsNone(cls)
+        cls = get_scheme_class("W8A8_DYNAMIC", "nonexistent")
+        self.assertIsNone(cls)
+
+    def test_all_linear_schemes_subclass_ascend_linear_scheme(self):
+        for (quant_type, layer_type), scheme_cls in _SCHEME_REGISTRY.items():
+            if layer_type == "linear":
+                self.assertTrue(
+                    issubclass(scheme_cls, AscendLinearScheme),
+                    f"{scheme_cls.__name__} for {quant_type}/{layer_type} should be subclass of AscendLinearScheme",
+                )
+
+    def test_all_moe_schemes_subclass_ascend_moe_scheme(self):
+        for (quant_type, layer_type), scheme_cls in _SCHEME_REGISTRY.items():
+            if layer_type == "moe":
+                self.assertTrue(
+                    issubclass(scheme_cls, AscendMoEScheme),
+                    f"{scheme_cls.__name__} for {quant_type}/{layer_type} should be subclass of AscendMoEScheme",
+                )
+
+    def test_registry_not_empty(self):
+        self.assertGreater(len(_SCHEME_REGISTRY), 0)