Skip to content

Commit c10e254

Browse files
authored
Torch compile test for python 3.8-3.11 linux only (#1629)
This should fix failure on with Python 3.12 validations: https://github.com/pytorch/builder/actions/runs/7064433251/job/19232483984#step:11:4859
1 parent 2a4c533 commit c10e254

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/smoke_test/smoke_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,8 @@ def smoke_test_cuda(package: str, runtime_error_check: str) -> None:
167167
print(f"torch cudnn: {torch.backends.cudnn.version()}")
168168
print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
169169

170-
# torch.compile is available only on Linux and python 3.8-3.10
171-
if sys.platform in ["linux", "linux2"] and (sys.version_info < (3, 11, 0) or channel != "release"):
170+
# torch.compile is available only on Linux and python 3.8-3.11
171+
if (sys.platform in ["linux", "linux2"]) and sys.version_info < (3, 12, 0):
172172
smoke_test_compile()
173173

174174
if runtime_error_check == "enabled":

0 commit comments

Comments
 (0)