Skip to content

Commit 44c3fd7

Browse files
committed
Fixing DORA imports
Summary: these imports were pointing at nothing Test Plan: python test/dora/test_dora_fusion.py Reviewers: Subscribers: Tasks: Tags:
1 parent 79e3366 commit 44c3fd7

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
@@ -9,14 +9,17 @@
99
TRITON_SUPPORTED_ACC_TYPES,
1010
SwizzleType,
1111
TritonInputPrecision,
12-
early_config_prune,
13-
estimate_matmul_time,
1412
get_compute_bound_configs,
15-
get_configs_io_bound,
16-
get_higher_dtype,
1713
swizzle_tile,
1814
to_tl_type,
1915
)
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+
)
2023
from .custom_autotune import autotune
2124

2225
logger = logging.getLogger(__name__)

0 commit comments

Comments
 (0)