Skip to content

Commit

Permalink
Re-enable compilation tests (pytorch#92333)
Browse files Browse the repository at this point in the history
As CUDA-11.5 is no longer supported, just remove the check

Fixes pytorch#69460

Pull Request resolved: pytorch#92333
Approved by: https://github.com/atalman
  • Loading branch information
malfet authored and pytorchmergebot committed Feb 6, 2023
1 parent 180adf8 commit a07d129
Showing 1 changed file with 0 additions and 11 deletions.
11 changes: 0 additions & 11 deletions test/run_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -1206,17 +1206,6 @@ def get_selected_tests(options):
WINDOWS_BLOCKLIST.append("jit")
WINDOWS_BLOCKLIST.append("jit_fuser")

# 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" 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")

selected_tests = exclude_tests(WINDOWS_BLOCKLIST, selected_tests, "on Windows")

elif TEST_WITH_ROCM:
Expand Down

0 comments on commit a07d129

Please sign in to comment.