frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reenable TestCppExtensionJIT on M1 (#84552)

Browse Source 
Works fine locally, let's see if it'll pass CI

Pull Request resolved: https://github.com/pytorch/pytorch/pull/84552
Approved by: https://github.com/kit1980
This commit is contained in:
Nikita Shulga
2022-09-06 17:49:29 +00:00
committed by PyTorch MergeBot
parent c771d73461
commit c794ee5cc1
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_cpp_extensions_jit.py
View File

@ -15,7 +15,7 @@ import torch
import torch.backends.cudnn
import torch.utils.cpp_extension
from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME
from torch.testing._internal.common_utils import gradcheck, skipIfSlowGradcheckEnv, IS_ARM64
from torch.testing._internal.common_utils import gradcheck, skipIfSlowGradcheckEnv
TEST_CUDA = torch.cuda.is_available() and CUDA_HOME is not None
@ -39,7 +39,6 @@ def remove_build_path():
# There's only one test that runs gracheck, run slow mode manually
@skipIfSlowGradcheckEnv
@unittest.skipIf(IS_ARM64, "Does not work on arm")
class TestCppExtensionJIT(common.TestCase):
"""Tests just-in-time cpp extensions.
Don't confuse this with the PyTorch JIT (aka TorchScript).