OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix lint errors in fbcode (#135614)

Browse Source 
Summary: Fixed a bunch of fbcode imports that happened to work but confused autodeps.  After this autodeps still suggests "improvements" to TARGETS (which breaks our builds) but at least it can find all the imports.

Test Plan:
```
fbpython fbcode/tools/build/buck/linters/lint_autoformat.py --linter=autodeps --default-exec-timeout=1800 -- fbcode/caffe2/TARGETS fbcode/caffe2/test/TARGETS
```
Before:
```
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "test_export" (from caffe2/test/export/testing.py:229) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See https://fbur$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "testing" (from caffe2/test/export/test_export.py:87) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See https://fburl$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "test_export" (from caffe2/test/export/test_serdes.py:9) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See https://fb$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "testing" (from caffe2/test/export/test_serdes.py:10) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See https://fburl$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "testing" (from caffe2/test/export/test_retraceability.py:7) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See https:$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "test_export" (from caffe2/test/export/test_retraceability.py:6) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See ht$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "testing" (from caffe2/test/export/test_export_nonstrict.py:7) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See http$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "test_export" (from caffe2/test/export/test_export_nonstrict.py:6) when processing rule "test_export". Please make sure it's listed in the srcs parameter of another rule. See $
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "test_export" (from caffe2/test/export/test_export_training_ir_to_run_decomp.py:8) when processing rule "test_export". Please make sure it's listed in the srcs parameter of an$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "testing" (from caffe2/test/export/test_export_training_ir_to_run_decomp.py:10) when processing rule "test_export". Please make sure it's listed in the srcs parameter of anoth$
ERROR while processing caffe2/test/TARGETS: Found "//python/typeshed_internal:typeshed_internal_library" owner for "cv2" but it is protected by visibility rules: [] (from caffe2/test/test_bundled_images.py:7) when processing rule "test_bundled_$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "caffe2.test.profiler_test_cpp_thread_lib" (from caffe2/test/profiler/test_cpp_thread.py:29) when processing rule "profiler_test_cpp_thread". Please make sure it's listed in t$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "torch._utils_internal.get_file_path_2" (from caffe2/test/test_custom_ops.py:23) when processing rule "custom_ops". Please make sure it's listed in the srcs parameter of anoth$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "torch._utils_internal.get_file_path_2" (from caffe2/test/test_public_bindings.py:13) when processing rule "public_bindings". Please make sure it's listed in the srcs paramete$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "torch._C._profiler.symbolize_tracebacks" (from caffe2/test/test_cuda.py:3348) when processing rule "test_cuda". Please make sure it's listed in the srcs parameter of another $
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for "torch._C._profiler.gather_traceback" (from caffe2/test/test_cuda.py:3348) when processing rule "test_cuda". Please make sure it's listed in the srcs parameter of another rule$
ERROR while processing caffe2/test/TARGETS: Cannot find an owner for include <torch/csrc/autograd/profiler_kineto.h> (from caffe2/test/profiler/test_cpp_thread.cpp:2) when processing profiler_test_cpp_thread_lib.  Some things to try:
```

Differential Revision: D62049222

Pull Request resolved: https://github.com/pytorch/pytorch/pull/135614
Approved by: https://github.com/oulgen, https://github.com/laithsakka
This commit is contained in:
Aaron Orenstein 2024-09-13 02:04:34 +00:00 committed by PyTorch MergeBot
parent bf68e16e94
commit 8c356ce3da
45 changed files with 120 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/export/test_export.py
View File

@ -84,7 +84,7 @@ except ImportError:
try:
from . import testing
except ImportError:
import testing
import testing # @manual=fbcode//caffe2/test:test_export-library
# The following import pattern matters as `test_export.export` is patched
# in other files (like test_export_nonstrict.py). `torch.export.export`
# will invalidate the patch.

4
test/export/test_export_nonstrict.py
View File

@ -3,8 +3,8 @@
try:
from . import test_export, testing
except ImportError:
import test_export
import testing
import test_export # @manual=fbcode//caffe2/test:test_export-library
import testing # @manual=fbcode//caffe2/test:test_export-library
from torch.export import export

4
test/export/test_export_training_ir_to_run_decomp.py
View File

@ -5,9 +5,9 @@ import torch
try:
from . import test_export, testing
except ImportError:
import test_export
import test_export # @manual=fbcode//caffe2/test:test_export-library
import testing
import testing # @manual=fbcode//caffe2/test:test_export-library
test_classes = {}

4
test/export/test_retraceability.py
View File

@ -3,8 +3,8 @@
try:
from . import test_export, testing
except ImportError:
import test_export
import testing
import test_export # @manual=fbcode//caffe2/test:test_export-library
import testing # @manual=fbcode//caffe2/test:test_export-library
from torch.export import export

4
test/export/test_serdes.py
View File

@ -6,8 +6,8 @@ import io
try:
from . import test_export, testing
except ImportError:
import test_export
import testing
import test_export # @manual=fbcode//caffe2/test:test_export-library
import testing # @manual=fbcode//caffe2/test:test_export-library
from torch.export import export, load, save

2
test/export/testing.py
View File

@ -226,7 +226,7 @@ def _make_fn_with_mocked_export(fn, mocked_export_fn):
try:
from . import test_export
except ImportError:
import test_export
import test_export # @manual=fbcode//caffe2/test:test_export-library
with patch(f"{test_export.__name__}.export", mocked_export_fn):
return fn(*args, **kwargs)

2
test/inductor/custom_ops.cpp
View File

@ -1,4 +1,4 @@
#include <torch/csrc/api/include/torch/types.h>
#include <torch/csrc/api/include/torch/types.h> // @manual=fbcode//caffe2:libtorch
#include <cstdint>
#include <iostream>

14
test/inductor/test_aot_inductor.py
View File

@ -45,7 +45,7 @@ from torch.utils import _pytree as pytree
if HAS_CUDA:
import triton
import triton # @manual
from torch.testing._internal.triton_utils import (
add_kernel,
@ -76,14 +76,20 @@ try:
)
from .test_torchinductor import copy_tests, requires_multigpu, TestFailure
except ImportError:
from test_aot_inductor_utils import AOTIRunnerUtil
from test_control_flow import (
from test_aot_inductor_utils import (
AOTIRunnerUtil, # @manual=fbcode//caffe2/test/inductor:aot_inductor_utils-library
)
from test_control_flow import ( # @manual=fbcode//caffe2/test/inductor:control_flow-library
CondModels,
prepend_counters,
prepend_predicates,
WhileLoopModels,
)
from test_torchinductor import copy_tests, requires_multigpu, TestFailure
from test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
copy_tests,
requires_multigpu,
TestFailure,
)
except (unittest.SkipTest, ImportError) as e:
if __name__ == "__main__":
sys.exit(0)

2
test/inductor/test_aot_inductor_utils.py
View File

@ -67,7 +67,7 @@ class AOTIRunnerUtil:
@staticmethod
def load_runner(device, so_path):
if IS_FBCODE:
from .fb import test_aot_inductor_model_runner_pybind
from .fb import test_aot_inductor_model_runner_pybind # @manual
with tempfile.TemporaryDirectory() as temp_dir:
# copy *.so file to a unique path just before loading

6
test/inductor/test_benchmark_fusion.py
View File

@ -20,7 +20,11 @@ sys.path.append(pytorch_test_dir)
import contextlib
import unittest
from inductor.test_torchinductor import check_model, check_model_cuda, copy_tests
from inductor.test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_cuda,
copy_tests,
)
from torch._inductor import config
from torch._inductor.scheduler import Scheduler

10
test/inductor/test_binary_folding.py
View File

@ -15,8 +15,14 @@ from torch.testing._internal.common_cuda import TEST_CUDNN
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(pytorch_test_dir)
from inductor.test_inductor_freezing import TestCase
from inductor.test_torchinductor import check_model, check_model_gpu, copy_tests
from inductor.test_inductor_freezing import (
TestCase, # @manual=fbcode//caffe2/test/inductor:inductor_freezing-library
)
from inductor.test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_gpu,
copy_tests,
)
from torch.testing._internal.common_utils import TEST_WITH_ASAN
from torch.testing._internal.inductor_utils import skipCUDAIf

2
test/inductor/test_ck_backend.py
View File

@ -43,7 +43,7 @@ class TestCKBackend(TestCase):
torch.random.manual_seed(1234)
try:
import ck4inductor
import ck4inductor # @manual
self.ck_dir = os.path.dirname(ck4inductor.__file__)
os.environ["TORCHINDUCTOR_CK_DIR"] = self.ck_dir

5
test/inductor/test_combo_kernels.py
View File

@ -20,7 +20,10 @@ try:
try:
from .test_torchinductor import check_model, check_model_cuda
except ImportError:
from test_torchinductor import check_model, check_model_cuda
from test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_cuda,
)
except (unittest.SkipTest, ImportError) as e:
sys.stderr.write(f"{type(e)}: {e}\n")
if __name__ == "__main__":

7
test/inductor/test_compiled_optimizers.py
View File

@ -270,7 +270,10 @@ try:
try:
from .test_torchinductor import check_model, check_model_gpu
except ImportError:
from test_torchinductor import check_model, check_model_gpu
from test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_gpu,
)
except (unittest.SkipTest, ImportError) as e:
sys.stderr.write(f"{type(e)}: {e}\n")
if __name__ == "__main__":
@ -835,7 +838,7 @@ class CompiledOptimizerTests(TestCase):
try:
from . import s429861_repro
except ImportError:
import s429861_repro
import s429861_repro # @manual
forward = s429861_repro.forward

2
test/inductor/test_coordinate_descent_tuner.py
View File

@ -12,7 +12,7 @@ from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_GPU
try:
import triton
import triton # @manual
except ImportError:
if __name__ == "__main__":
sys.exit(0)

10
test/inductor/test_cpu_cpp_wrapper.py
View File

@ -28,11 +28,11 @@ try:
test_torchinductor_dynamic_shapes,
)
except ImportError:
import test_cpu_repro
import test_cpu_select_algorithm
import test_mkldnn_pattern_matcher
import test_torchinductor
import test_torchinductor_dynamic_shapes
import test_cpu_repro # @manual=fbcode//caffe2/test/inductor:test_cpu_repro-library
import test_cpu_select_algorithm # @manual=fbcode//caffe2/test/inductor:cpu_select_algorithm_cpu-library
import test_mkldnn_pattern_matcher # @manual
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
import test_torchinductor_dynamic_shapes # @manual=fbcode//caffe2/test/inductor:test_inductor-library_dynamic_shapes
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

2
test/inductor/test_cpu_repro.py
View File

@ -42,7 +42,7 @@ try:
try:
from . import test_torchinductor
except ImportError:
import test_torchinductor
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

4
test/inductor/test_cpu_select_algorithm.py
View File

@ -39,8 +39,8 @@ try:
try:
from . import test_cpu_repro, test_torchinductor
except ImportError:
import test_cpu_repro
import test_torchinductor
import test_cpu_repro # @manual=fbcode//caffe2/test/inductor:test_cpu_repro-library
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

10
test/inductor/test_cuda_cpp_wrapper.py
View File

@ -26,11 +26,11 @@ try:
except ImportError:
import test_combo_kernels
import test_foreach
import test_pattern_matcher
import test_select_algorithm
import test_torchinductor
import test_torchinductor_dynamic_shapes
import test_foreach # @manual=fbcode//caffe2/test/inductor:foreach-library
import test_pattern_matcher # @manual=fbcode//caffe2/test/inductor:pattern_matcher-library
import test_select_algorithm # @manual=fbcode//caffe2/test/inductor:select_algorithm-library
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
import test_torchinductor_dynamic_shapes # @manual=fbcode//caffe2/test/inductor:test_inductor-library_dynamic_shapes
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

6
test/inductor/test_cuda_repro.py
View File

@ -38,15 +38,15 @@ from torch.testing._internal.inductor_utils import skipCUDAIf
try:
try:
import triton
from triton import language as tl
import triton # @manual
from triton import language as tl # @manual
except ImportError:
raise unittest.SkipTest("requires triton") # noqa: B904
try:
from . import test_torchinductor
except ImportError:
import test_torchinductor
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

6
test/inductor/test_cudagraph_trees_expandable_segments.py
View File

@ -18,12 +18,14 @@ if HAS_CUDA and not TEST_WITH_ASAN:
try:
from .test_cudagraph_trees import CudaGraphTreeTests
except ImportError:
from test_cudagraph_trees import CudaGraphTreeTests # noqa: F401
from test_cudagraph_trees import ( # noqa: F401 # @manual=fbcode//caffe2/test/inductor:cudagraph_trees-library
CudaGraphTreeTests,
)
REPO_ROOT = pathlib.Path(__file__).resolve().parent.parent.parent
sys.path.insert(0, str(REPO_ROOT))
from tools.stats.import_test_stats import get_disabled_tests
from tools.stats.import_test_stats import get_disabled_tests # @manual
# Make sure to remove REPO_ROOT after import is done

2
test/inductor/test_debug_trace.py
View File

@ -18,7 +18,7 @@ try:
try:
from . import test_torchinductor
except ImportError:
import test_torchinductor
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

4
test/inductor/test_efficient_conv_bn_eval.py
View File

@ -23,7 +23,9 @@ from torch.testing._internal.inductor_utils import HAS_CPU, HAS_CUDA
importlib.import_module("functorch")
importlib.import_module("filelock")
from inductor.test_torchinductor import copy_tests
from inductor.test_torchinductor import (
copy_tests, # @manual=fbcode//caffe2/test/inductor:test_inductor-library
)
class ConvOp(nn.Module):

4
test/inductor/test_extension_backend.py
View File

@ -11,7 +11,7 @@ from torch._C import FileCheck
try:
from extension_backends.cpp.extension_codegen_backend import (
from extension_backends.cpp.extension_codegen_backend import ( # @manual=fbcode//caffe2/test/inductor/extension_backends:extension_codegen_backend # noqa: B950
ExtensionCppWrapperCodegen,
ExtensionScheduling,
ExtensionWrapperCodegen,
@ -38,7 +38,7 @@ try:
try:
from . import test_torchinductor
except ImportError:
import test_torchinductor
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

5
test/inductor/test_foreach.py
View File

@ -21,7 +21,10 @@ try:
try:
from .test_torchinductor import check_model, check_model_cuda
except ImportError:
from test_torchinductor import check_model, check_model_cuda
from test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_cuda,
)
except (unittest.SkipTest, ImportError) as e:
sys.stderr.write(f"{type(e)}: {e}\n")
if __name__ == "__main__":

4
test/inductor/test_halide.py
View File

@ -27,7 +27,7 @@ if IS_WINDOWS and IS_CI:
raise unittest.SkipTest("requires sympy/functorch/filelock")
try:
import halide
import halide # @manual
HAS_HALIDE = halide is not None
except ImportError:
@ -37,7 +37,7 @@ except ImportError:
try:
from . import test_torchinductor
except ImportError:
import test_torchinductor
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
make_halide = config.patch(

6
test/inductor/test_inductor_freezing.py
View File

@ -23,7 +23,11 @@ from torch.testing._internal.common_utils import skipIfRocm
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(pytorch_test_dir)
from inductor.test_torchinductor import check_model, check_model_cuda, copy_tests
from inductor.test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_cuda,
copy_tests,
)
from torch.testing._internal.common_utils import TEST_WITH_ASAN, TEST_WITH_ROCM

4
test/inductor/test_inplacing_pass.py
View File

@ -46,8 +46,8 @@ def sin_cos(x: torch.Tensor, out_sin: torch.Tensor, out_cos: torch.Tensor) -> No
if HAS_GPU:
import triton
import triton.language as tl
import triton # @manual
import triton.language as tl # @manual
@triton.jit
def sin_kernel(

4
test/inductor/test_memory_planning.py
View File

@ -84,7 +84,9 @@ class TestMemoryPlanning(TestCase):
try:
from .test_aot_inductor import AOTIRunnerUtil
except ImportError:
from test_aot_inductor import AOTIRunnerUtil
from test_aot_inductor import (
AOTIRunnerUtil, # @manual=fbcode//caffe2/test/inductor:test_aot_inductor-library
)
f, args = self._generate(device="cuda")
dim0_x = Dim("dim0_x", min=1, max=2048)

4
test/inductor/test_perf.py
View File

@ -32,8 +32,8 @@ from torch.testing._internal.triton_utils import HAS_CUDA, requires_cuda
if HAS_CUDA:
import triton
import triton.language as tl
import triton # @manual
import triton.language as tl # @manual
from torch.testing._internal.triton_utils import add_kernel

2
test/inductor/test_profiler.py
View File

@ -172,7 +172,7 @@ class DynamoProfilerTests(torch._inductor.test_case.TestCase):
@unittest.skipIf(not HAS_TRITON, "requires cuda & triton")
def test_inductor_profiling_triton_hooks(self):
from triton.compiler import CompiledKernel
from triton.compiler import CompiledKernel # @manual
hooks_called = {"enter": False, "exit": False}

4
test/inductor/test_torchinductor_codegen_dynamic_shapes.py
View File

@ -20,14 +20,14 @@ importlib.import_module("filelock")
# Make the helper files in test/ importable
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(pytorch_test_dir)
from inductor.test_torchinductor import (
from inductor.test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
CommonTemplate,
copy_tests,
run_and_get_cpp_code,
run_and_get_triton_code,
TestFailure,
)
from inductor.test_torchinductor_dynamic_shapes import (
from inductor.test_torchinductor_dynamic_shapes import ( # @manual
make_dynamic_cls,
test_failures as dynamic_shapes_test_failures,
)

2
test/inductor/test_torchinductor_dynamic_shapes.py
View File

@ -39,7 +39,7 @@ from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CPU, HAS_GPU
# Make the helper files in test/ importable
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(pytorch_test_dir)
from inductor.test_torchinductor import (
from inductor.test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_gpu,
CommonTemplate,

5
test/inductor/test_torchinductor_opinfo.py
View File

@ -49,7 +49,10 @@ try:
try:
from .test_torchinductor import check_model, check_model_gpu
except ImportError:
from test_torchinductor import check_model, check_model_gpu
from test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model,
check_model_gpu,
)
except (unittest.SkipTest, ImportError) as e:
sys.stderr.write(f"{type(e)}: {e}\n")
if __name__ == "__main__":

8
test/inductor/test_triton_extension_backend.py
View File

@ -10,8 +10,10 @@ import torch.utils.cpp_extension
try:
from extension_backends.triton.device_interface import DeviceInterface
from extension_backends.triton.extension_codegen_backend import (
from extension_backends.triton.device_interface import (
DeviceInterface, # @manual=fbcode//caffe2/test/inductor/extension_backends:extension_codegen_backend
)
from extension_backends.triton.extension_codegen_backend import ( # @manual=fbcode//caffe2/test/inductor/extension_backends:extension_codegen_backend # noqa: B950
CPUDeviceOpOverrides,
ExtensionScheduling,
ExtensionWrapperCodegen,
@ -41,7 +43,7 @@ try:
try:
from . import test_torchinductor
except ImportError:
import test_torchinductor
import test_torchinductor # @manual=fbcode//caffe2/test/inductor:test_inductor-library
except unittest.SkipTest:
if __name__ == "__main__":
sys.exit(0)

6
test/inductor/test_triton_heuristics.py
View File

@ -9,8 +9,8 @@ from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_GPU
try:
import triton # noqa: F401
import triton.language as tl
import triton # noqa: F401 # @manual
import triton.language as tl # @manual
except ImportError:
if __name__ == "__main__":
sys.exit(0)
@ -88,7 +88,7 @@ class TestTritonHeuristics(TestCase):
self._test_artificial_zgrid()
def _get_cos_kernel_caching_autotuner_args(self):
from triton.compiler.compiler import AttrsDescriptor
from triton.compiler.compiler import AttrsDescriptor # @manual
@triton.jit
def triton_(in_ptr0, out_ptr0, xnumel, XBLOCK: tl.constexpr):

8
test/inductor/test_triton_kernels.py
View File

@ -36,12 +36,12 @@ if HAS_GPU:
if not TEST_WITH_ROCM:
if HAS_CUDA:
from triton.language.extra.cuda.libdevice import (
from triton.language.extra.cuda.libdevice import ( # @manual
fast_dividef,
fast_dividef as my_fast_dividef,
)
elif HAS_XPU:
from triton.language.extra.intel.libdevice import (
from triton.language.extra.intel.libdevice import ( # @manual
fast_dividef,
fast_dividef as my_fast_dividef,
)
@ -2543,8 +2543,8 @@ class CustomOpTests(torch._inductor.test_case.TestCase):
@requires_gpu
def test_capture_triton_disabled_in_triton_op(self):
import triton
import triton.language as tl
import triton # @manual
import triton.language as tl # @manual
@triton.jit
def add_kernel(

5
test/inductor/test_xpu_basic.py
View File

@ -20,7 +20,10 @@ importlib.import_module("filelock")
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(pytorch_test_dir)
from inductor.test_torchinductor import check_model_gpu, TestCase
from inductor.test_torchinductor import ( # @manual=fbcode//caffe2/test/inductor:test_inductor-library
check_model_gpu,
TestCase,
)
# TODO: Remove this file.

2
test/profiler/test_cpp_thread.cpp
View File

@ -1,5 +1,5 @@
#include <torch/csrc/autograd/profiler_kineto.h>
#include <torch/csrc/autograd/profiler_kineto.h> // @manual
#include <torch/torch.h>
#include <string>

2
test/profiler/test_cpp_thread.py
View File

@ -26,7 +26,7 @@ def is_fbcode():
if is_fbcode():
import caffe2.test.profiler_test_cpp_thread_lib as cpp
import caffe2.test.profiler_test_cpp_thread_lib as cpp # @manual=//caffe2/test:profiler_test_cpp_thread_lib
else:
# cpp extensions use relative paths. Those paths are relative to
# this file, so we'll change the working directory temporarily

0
test/profiler/test_cpp_thread_lib.pyi Normal file
View File

2
test/test_bundled_images.py
View File

@ -4,7 +4,7 @@
import io
import cv2
import cv2 # @manual
import torch
import torch.utils.bundled_inputs

2
test/test_cuda.py
View File

@ -3345,7 +3345,7 @@ class TestCudaMallocAsync(TestCase):
@unittest.skipIf(IS_ARM64 or not IS_LINUX, "x86 linux only cpp unwinding")
def test_direct_traceback(self):
from torch._C._profiler import gather_traceback, symbolize_tracebacks
from torch._C._profiler import gather_traceback, symbolize_tracebacks # @manual
c = gather_traceback(True, True, True)
(r,) = symbolize_tracebacks([c])

2
test/test_custom_ops.py
View File

@ -20,7 +20,7 @@ from functorch import make_fx
from torch import Tensor
from torch._custom_op.impl import CustomOp, infer_schema
from torch._library.infer_schema import tuple_to_list
from torch._utils_internal import get_file_path_2
from torch._utils_internal import get_file_path_2 # @manual
from torch.testing._internal import custom_op_db
from torch.testing._internal.common_cuda import TEST_CUDA
from torch.testing._internal.common_device_type import (

2
test/test_public_bindings.py
View File

@ -10,7 +10,7 @@ import unittest
from typing import Callable
import torch
from torch._utils_internal import get_file_path_2
from torch._utils_internal import get_file_path_2 # @manual
from torch.testing._internal.common_utils import (
IS_JETSON,
IS_MACOS,