Skip to content

Commit ac832b0

Browse files
authored
Fixing DORA imports (#1795)
* Fixing DORA imports Summary: these imports were pointing at nothing Test Plan: python test/dora/test_dora_fusion.py Reviewers: Subscribers: Tasks: Tags: * fixing lint issues Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
1 parent b9c51b7 commit ac832b0

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

torchao/prototype/dora/kernels/matmul.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,19 @@
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,
1017
SwizzleType,
1118
TritonInputPrecision,
12-
early_config_prune,
13-
estimate_matmul_time,
1419
get_compute_bound_configs,
15-
get_configs_io_bound,
16-
get_higher_dtype,
1720
swizzle_tile,
1821
to_tl_type,
1922
)

0 commit comments

Comments
 (0)