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
pytorch · Sep 14, 2023 · 12a6ea5 · 12a6ea5
1 parent 0c1d107
commit 12a6ea5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -129,7 +129,9 @@ def smoke_test_cuda(package: str, runtime_error_check: str) -> None:
         print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
 
         # torch.compile is available only on Linux and python 3.8-3.10
-        if (sys.platform == "linux" or sys.platform == "linux2") and sys.version_info < (3, 11, 0):
+        if (sys.platform == "linux" or sys.platform == "linux2") and sys.version_info < (3, 11, 0) and channel == "release":
+            smoke_test_compile()
+        elif (sys.platform == "linux" or sys.platform == "linux2") and channel != "release":
             smoke_test_compile()
 
         if(runtime_error_check == "enabled"):