Expose ops (handy for tests etc.)

ext-torch/__init__.py

@@ -13,6 +13,7 @@ from .marlin import (
     marlin_qqq_gemm,
     marlin_gemm,
 )
+from ._ops import ops
 
 __all__ = [
     "awq_marlin_repack",
@@ -25,6 +26,7 @@ __all__ = [
     "gptq_marlin_repack",
     "marlin_gemm",
     "marlin_qqq_gemm",
+    "ops",
     "scaled_fp8_quant",
     "scaled_int8_quant",
 ]