We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8a92363 commit b0ab8bcCopy full SHA for b0ab8bc
1 file changed
deep_gemm/__init__.py
@@ -13,7 +13,7 @@
13
except ImportError:
14
pass
15
16
-import deep_gemm_cpp # noqa: F401 # Registers ops into torch.ops without touching CUDA
+from . import deep_gemm_cpp # noqa: F401 # Registers ops into torch.ops without touching CUDA
17
18
19
def _find_cuda_home() -> str:
0 commit comments