Skip to content

Commit 29bf2f9

Browse files
committed
Refactor import mm to import ops.ninetoothed.kernels.mm as mm in bmm.py
1 parent db86285 commit 29bf2f9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

bmm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import torch
33
from ninetoothed import Symbol, Tensor
44

5-
import mm
5+
import ops.ninetoothed.kernels.mm as mm
66

77
BLOCK_SIZE_M = Symbol("BLOCK_SIZE_M", meta=True)
88
BLOCK_SIZE_N = Symbol("BLOCK_SIZE_N", meta=True)

0 commit comments

Comments
 (0)