Torch compile test for python 3.8-3.11 linux only

pytorch · Dec 1, 2023 · 7317456 · 7317456
1 parent 4db3d68
commit 7317456
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -167,8 +167,8 @@ def smoke_test_cuda(package: str, runtime_error_check: str) -> None:
         print(f"torch cudnn: {torch.backends.cudnn.version()}")
         print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
 
-        # torch.compile is available only on Linux and python 3.8-3.10
-        if sys.platform in ["linux", "linux2"] and (sys.version_info < (3, 11, 0) or channel != "release"):
+        # torch.compile is available only on Linux and python 3.8-3.11
+        if sys.platform in ["linux", "linux2"] and sys.version_info =< (3, 11, 0):
             smoke_test_compile()
 
         if runtime_error_check == "enabled":