Skip to content

Commit 12a6ea5

Browse files
authored
Enable torch compile for python 3.11 smoke tests (#1534)
* Enable torch compile for python 3.11 smoke tests * Make sure release is covered * Fix typo
1 parent 0c1d107 commit 12a6ea5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/smoke_test/smoke_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,9 @@ def smoke_test_cuda(package: str, runtime_error_check: str) -> None:
129129
print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
130130

131131
# torch.compile is available only on Linux and python 3.8-3.10
132-
if (sys.platform == "linux" or sys.platform == "linux2") and sys.version_info < (3, 11, 0):
132+
if (sys.platform == "linux" or sys.platform == "linux2") and sys.version_info < (3, 11, 0) and channel == "release":
133+
smoke_test_compile()
134+
elif (sys.platform == "linux" or sys.platform == "linux2") and channel != "release":
133135
smoke_test_compile()
134136

135137
if(runtime_error_check == "enabled"):

0 commit comments

Comments
 (0)