Skip to content

Commit

Permalink
Reenable aot tests on windows for cuda 11.7 and up (#87193) (#87307)
Browse files Browse the repository at this point in the history
Reenable aot tests on windows for cuda 11.7 and up

Issue: #69460 seems to be mitigated in CUDA 11.7 hence re-enable this test

cc @peterjc123 @mszhanyi @skyline75489 @nbcsm
Pull Request resolved: #87193
Approved by: https://github.com/malfet
  • Loading branch information
atalman committed Oct 20, 2022
1 parent 9e3df49 commit 3ca59f4
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion test/run_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 3ca59f4

Please sign in to comment.