diff --git a/.gitignore b/.gitignore index 7557c564a6d..ba582c61c4d 100644 --- a/.gitignore +++ b/.gitignore @@ -62,9 +62,7 @@ test/forward_backward_compatibility/nightly_schemas.txt dropout_model.pt test/generated_type_hints_smoketest.py test/htmlcov -test/cpp_extensions/install/ -test/cpp_extensions/open_registration_extension/install -test/cpp_extensions/libtorch_agnostic_extension/install +test/cpp_extensions/**/install test/kernel.errors.txt third_party/build/ third_party/nccl/ diff --git a/test/test_cpp_extensions_aot.py b/test/test_cpp_extensions_aot.py index c9c3129adfa..ee8dfac2966 100644 --- a/test/test_cpp_extensions_aot.py +++ b/test/test_cpp_extensions_aot.py @@ -3,7 +3,6 @@ import os import re import subprocess -import sys import unittest from itertools import repeat from pathlib import Path @@ -16,7 +15,6 @@ import torch.utils.cpp_extension from torch.testing._internal.common_cuda import TEST_CUDA from torch.testing._internal.common_utils import ( IS_WINDOWS, - shell, skipIfTorchDynamo, TEST_XPU, xfailIfTorchDynamo, @@ -217,16 +215,6 @@ class TestCppExtensionAOT(common.TestCase): missing_symbols = subprocess.check_output(["nm", "-u", so_file]).decode("utf-8") self.assertFalse("Py" in missing_symbols) - # finally, clean up the folder - cmd = [sys.executable, "setup.py", "clean"] - return_code = shell( - cmd, - cwd=os.path.join("cpp_extensions", "python_agnostic_extension"), - env=os.environ.copy(), - ) - if return_code != 0: - return return_code - @unittest.skipIf(not TEST_CUDA, "some aspects of this test require CUDA") def test_libtorch_agnostic(self): import libtorch_agnostic