Skip to content

Commit b0ab8bc

Browse files
authored
fix sgl import
1 parent 8a92363 commit b0ab8bc

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

deep_gemm/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
except ImportError:
1414
pass
1515

16-
import deep_gemm_cpp # noqa: F401 # Registers ops into torch.ops without touching CUDA
16+
from . import deep_gemm_cpp # noqa: F401 # Registers ops into torch.ops without touching CUDA
1717

1818

1919
def _find_cuda_home() -> str:

0 commit comments

Comments
 (0)