Skip to content

Commit 17ba37d

Browse files
authored
Correct torchao check for TensorCoreTiledLayout (#1886)
1 parent 1e5f0d5 commit 17ba37d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchtune/utils/_import_guard.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
_USE_NEW_TENSOR_CORE_TILED_LAYOUT_API = _is_fbcode() or (
2121
not _is_fbcode()
2222
and (
23-
("dev" not in torchao_version and torchao_version >= "0.6.0")
23+
("dev" not in torchao_version and torchao_version >= "0.7.0")
2424
or (
2525
"dev" in torchao_version
2626
and _nightly_version_ge(torchao_version, "2024-10-10")

0 commit comments

Comments
 (0)