Skip to content

Commit b04d4cd

Browse files
committed
fixing lint issues
Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
1 parent 44c3fd7 commit b04d4cd

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

torchao/prototype/dora/kernels/matmul.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,13 @@
44
import triton
55
import triton.language as tl
66

7+
from torchao.prototype.common.triton.matmul import (
8+
early_config_prune,
9+
estimate_matmul_time,
10+
get_configs_io_bound,
11+
get_higher_dtype,
12+
)
13+
714
from .common import (
815
MATMUL_HEURISTICS,
916
TRITON_SUPPORTED_ACC_TYPES,
@@ -13,13 +20,6 @@
1320
swizzle_tile,
1421
to_tl_type,
1522
)
16-
from torchao.prototype.common.triton.matmul import (
17-
get_configs_io_bound,
18-
get_higher_dtype,
19-
early_config_prune,
20-
estimate_matmul_time,
21-
get_higher_dtype,
22-
)
2323
from .custom_autotune import autotune
2424

2525
logger = logging.getLogger(__name__)

0 commit comments

Comments
 (0)