Expose ops (handy for tests etc.)
Browse files- ext-torch/__init__.py +2 -0
ext-torch/__init__.py
CHANGED
|
@@ -13,6 +13,7 @@ from .marlin import (
|
|
| 13 |
marlin_qqq_gemm,
|
| 14 |
marlin_gemm,
|
| 15 |
)
|
|
|
|
| 16 |
|
| 17 |
__all__ = [
|
| 18 |
"awq_marlin_repack",
|
|
@@ -25,6 +26,7 @@ __all__ = [
|
|
| 25 |
"gptq_marlin_repack",
|
| 26 |
"marlin_gemm",
|
| 27 |
"marlin_qqq_gemm",
|
|
|
|
| 28 |
"scaled_fp8_quant",
|
| 29 |
"scaled_int8_quant",
|
| 30 |
]
|
|
|
|
| 13 |
marlin_qqq_gemm,
|
| 14 |
marlin_gemm,
|
| 15 |
)
|
| 16 |
+
from ._ops import ops
|
| 17 |
|
| 18 |
__all__ = [
|
| 19 |
"awq_marlin_repack",
|
|
|
|
| 26 |
"gptq_marlin_repack",
|
| 27 |
"marlin_gemm",
|
| 28 |
"marlin_qqq_gemm",
|
| 29 |
+
"ops",
|
| 30 |
"scaled_fp8_quant",
|
| 31 |
"scaled_int8_quant",
|
| 32 |
]
|