From 73174566d0fe6530544cb2087eb893db4a30f86f Mon Sep 17 00:00:00 2001 From: atalman Date: Fri, 1 Dec 2023 13:05:26 -0800 Subject: [PATCH 1/3] 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": From 5e54cecbe1adaecafd5dd9a3be2fa20966b8c839 Mon Sep 17 00:00:00 2001 From: atalman Date: Fri, 1 Dec 2023 13:17:46 -0800 Subject: [PATCH 2/3] e9999 --- test/smoke_test/smoke_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index 67444aa36..af7f57b2f 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -168,7 +168,7 @@ 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.11 - if sys.platform in ["linux", "linux2"] and sys.version_info =< (3, 11, 0): + if sys.platform in ["linux", "linux2"] and sys.version_info =< (3, 11, 0): # noqa: E999 smoke_test_compile() if runtime_error_check == "enabled": From a5037c45a31af5f1b0b52abad9b162b90a0bd808 Mon Sep 17 00:00:00 2001 From: atalman Date: Fri, 1 Dec 2023 13:20:53 -0800 Subject: [PATCH 3/3] test --- test/smoke_test/smoke_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index af7f57b2f..3b5b18c35 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -168,7 +168,7 @@ 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.11 - if sys.platform in ["linux", "linux2"] and sys.version_info =< (3, 11, 0): # noqa: E999 + if (sys.platform in ["linux", "linux2"]) and sys.version_info < (3, 12, 0): smoke_test_compile() if runtime_error_check == "enabled":