Skip to content

Commit 8f8a195

Browse files
authored
Revert "Update smoke_test.py - temporary disable torch.compile tests" (#8765)
1 parent e21cd0b commit 8f8a195

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

test/smoke_test.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -102,9 +102,8 @@ def main() -> None:
102102
smoke_test_torchvision_resnet50_classify("cuda")
103103

104104
# TODO: remove once pytorch/pytorch#110436 is resolved
105-
# Temporary Disabling compile test. Untill triton with Manylinux2014 is available
106-
# if sys.version_info < (3, 12, 0):
107-
# smoke_test_compile()
105+
if sys.version_info < (3, 12, 0):
106+
smoke_test_compile()
108107

109108
if torch.backends.mps.is_available():
110109
smoke_test_torchvision_resnet50_classify("mps")

0 commit comments

Comments
 (0)