Export `ScalarType`/`scalartypes`

ext-torch/quantization/__init__.py

@@ -13,9 +13,15 @@ from .marlin import (
     marlin_qqq_gemm,
     marlin_gemm,
 )
+from .scalar_type import (
+    ScalarType,
+    scalar_types,
+)
 from ._ops import ops
 
+
 __all__ = [
+    "ScalarType",
     "awq_marlin_repack",
     "cutlass_scaled_mm",
     "cutlass_scaled_mm_azp",
@@ -27,6 +33,7 @@ __all__ = [
     "marlin_gemm",
     "marlin_qqq_gemm",
     "ops",
+    "scalar_types",
     "scaled_fp8_quant",
     "scaled_int8_quant",
 ]