We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
import mm
import ops.ninetoothed.kernels.mm as mm
bmm.py
1 parent db86285 commit 29bf2f9Copy full SHA for 29bf2f9
1 file changed
bmm.py
@@ -2,7 +2,7 @@
2
import torch
3
from ninetoothed import Symbol, Tensor
4
5
-import mm
+import ops.ninetoothed.kernels.mm as mm
6
7
BLOCK_SIZE_M = Symbol("BLOCK_SIZE_M", meta=True)
8
BLOCK_SIZE_N = Symbol("BLOCK_SIZE_N", meta=True)
0 commit comments