From 3ca59f4b8aff32fbad3fcd13b812b5335047b8ce Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Thu, 20 Oct 2022 13:04:01 -0400 Subject: [PATCH] Reenable aot tests on windows for cuda 11.7 and up (#87193) (#87307) Reenable aot tests on windows for cuda 11.7 and up Issue: https://github.com/pytorch/pytorch/issues/69460 seems to be mitigated in CUDA 11.7 hence re-enable this test cc @peterjc123 @mszhanyi @skyline75489 @nbcsm Pull Request resolved: https://github.com/pytorch/pytorch/pull/87193 Approved by: https://github.com/malfet --- test/run_test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/run_test.py b/test/run_test.py index 8dff95ed6d90e..f7c80f3f0a6d3 100755 --- a/test/run_test.py +++ b/test/run_test.py @@ -1063,7 +1063,11 @@ def get_selected_tests(options): # This is exception that's caused by this issue https://github.com/pytorch/pytorch/issues/69460 # This below code should be removed once this issue is solved - if torch.version.cuda is not None and LooseVersion(torch.version.cuda) >= "11.5": + if ( + torch.version.cuda is not None and + LooseVersion(torch.version.cuda) >= "11.5" and + LooseVersion(torch.version.cuda) <= "11.6" + ): WINDOWS_BLOCKLIST.append("test_cpp_extensions_aot") WINDOWS_BLOCKLIST.append("test_cpp_extensions_aot_ninja") WINDOWS_BLOCKLIST.append("test_cpp_extensions_aot_no_ninja")