From 73174566d0fe6530544cb2087eb893db4a30f86f Mon Sep 17 00:00:00 2001 From: atalman Date: Fri, 1 Dec 2023 13:05:26 -0800 Subject: [PATCH] Torch compile test for python 3.8-3.11 linux only --- test/smoke_test/smoke_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index a4cd6dff2..67444aa36 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, 11, 0): smoke_test_compile() if runtime_error_check == "enabled":