diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index a4cd6dff2..3b5b18c35 100644 --- 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, 12, 0): smoke_test_compile() if runtime_error_check == "enabled":