Home
last modified time | relevance | path

Searched refs:cpp_extension (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/
H A Dtest_cpp_extensions_jit.py17 import torch.utils.cpp_extension
20 from torch.utils.cpp_extension import (
39 default_build_root = torch.utils.cpp_extension.get_default_build_root()
77 module = torch.utils.cpp_extension.load(
110 module = torch.utils.cpp_extension.load(
131 module = torch.utils.cpp_extension.load(
198 p = mp.Process(target=torch.utils.cpp_extension.load, kwargs=params)
208 torch.utils.cpp_extension.load(**params)
283 module = torch.utils.cpp_extension.load(
306 module = torch.utils.cpp_extension.load_inline(
[all …]
H A Dtest_cpp_extensions_aot.py12 import torch.utils.cpp_extension
28 cpp_extension = pytest.importorskip("torch_test_cpp_extension.cpp") variable
32 import torch_test_cpp_extension.cpp as cpp_extension namespace
56 z = cpp_extension.sigmoid_add(x, y)
60 str(torch.float32), str(cpp_extension.get_math_type(torch.half))
64 mm = cpp_extension.MatrixMultiplier(4, 8)
71 mm = cpp_extension.MatrixMultiplier(4, 8)
142 has_value = cpp_extension.function_taking_optional(torch.ones(5))
144 has_value = cpp_extension.function_taking_optional(None)
240 cpp_extension.get_complex,
[all …]
H A Dtest_cpp_extensions_stream_and_event.py11 import torch.utils.cpp_extension
21 from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME
33 default_build_root = torch.utils.cpp_extension.get_default_build_root()
78 cls.module = torch.utils.cpp_extension.load(
H A Dtest_cpp_extensions_mtia_backend.py11 import torch.utils.cpp_extension
20 from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME
32 default_build_root = torch.utils.cpp_extension.get_default_build_root()
68 cls.module = torch.utils.cpp_extension.load(
H A Dtest_cpp_extensions_open_device_registration.py17 import torch.utils.cpp_extension
26 from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME
37 default_build_root = torch.utils.cpp_extension.get_default_build_root()
103 cls.module = torch.utils.cpp_extension.load(
H A Dtest_utils.py20 import torch.utils.cpp_extension
973 exec_path = torch.utils.cpp_extension.load(
1194 self.assertTrue(torch.utils.cpp_extension.check_compiler_ok_for_platform("c++"))
1197 self.assertTrue(torch.utils.cpp_extension.check_compiler_ok_for_platform("cc"))
H A Dtest_dispatch.py9 import torch.utils.cpp_extension
962 module = torch.utils.cpp_extension.load(
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Dcompile_command.py14 from torch.utils import cpp_extension
19 else cpp_extension._join_rocm_home("include")
24 config.rocm.ck_dir or cpp_extension._join_rocm_home("composable_kernel"),
31 from torch.utils import cpp_extension
36 else cpp_extension._join_rocm_home("lib")
41 else cpp_extension._join_rocm_home("hip", "lib")
89 from torch.utils import cpp_extension
92 cpp_extension._join_rocm_home("llvm", "bin", "clang")
/aosp_15_r20/external/pytorch/tools/dynamo/
H A Dverify_dynamo.py36 from torch.utils import cpp_extension
38 CUDA_HOME = cpp_extension._find_cuda_home()
40 raise VerifyDynamoError(cpp_extension.CUDA_NOT_FOUND_MESSAGE)
46 .decode(*cpp_extension.SUBPROCESS_DECODE_ARGS)
58 from torch.utils import cpp_extension
60 ROCM_HOME = cpp_extension._find_rocm_home()
70 .decode(*cpp_extension.SUBPROCESS_DECODE_ARGS)
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dcpp_jit.py13 from torch.utils import cpp_extension
96 COMPAT_CALLGRIND_BINDINGS = cpp_extension.load(
144 return cpp_extension.load(
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_extension_backend.py9 import torch.utils.cpp_extension
54 default_build_root = torch.utils.cpp_extension.get_default_build_root()
73 cls.module = torch.utils.cpp_extension.load(
H A Dtest_compiled_autograd.py1591 module = torch.utils.cpp_extension.load_inline(
1640 module = torch.utils.cpp_extension.load_inline(
1708 module = torch.utils.cpp_extension.load_inline(
1805 module = torch.utils.cpp_extension.load_inline(
1862 module = torch.utils.cpp_extension.load_inline(
1921 module = torch.utils.cpp_extension.load_inline(
1979 module = torch.utils.cpp_extension.load_inline(
2070 module = torch.utils.cpp_extension.load_inline(
2342 module = torch.utils.cpp_extension.load_inline(
H A Dtest_triton_extension_backend.py9 import torch.utils.cpp_extension
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_cpp_thread.py9 import torch.utils.cpp_extension
14 default_build_root = torch.utils.cpp_extension.get_default_build_root()
36 cpp = torch.utils.cpp_extension.load(
/aosp_15_r20/external/pytorch/docs/source/
H A Dcpp_extension.rst1 torch.utils.cpp_extension
4 .. currentmodule:: torch.utils.cpp_extension
H A Dindex.rst120 torch.utils.cpp_extension <cpp_extension>
/aosp_15_r20/external/pytorch/test/cpp_api_parity/
H A Dutils.py9 import torch.utils.cpp_extension
131 cpp_module = torch.utils.cpp_extension.load_inline(
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/
H A Dbenchmark_core.py15 import torch.utils.cpp_extension as cpp_extension # noqa: F401 namespace
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt_extension/
H A Dsetup.py3 from torch.utils.cpp_extension import BuildExtension, CppExtension
/aosp_15_r20/external/pytorch/test/cpp_extensions/no_python_abi_suffix_test/
H A Dsetup.py3 from torch.utils.cpp_extension import BuildExtension, CppExtension
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcpp_builder.py746 from torch.utils.cpp_extension import _TORCH_PATH, TORCH_LIB_PATH
1191 from torch.utils import cpp_extension
1193 include_dirs = cpp_extension.include_paths(cuda)
1194 libraries_dirs = cpp_extension.library_paths(cuda)
H A Dcodecache.py1699 from torch.utils import cpp_extension
1719 ipaths = cpp_extension.include_paths(cuda) + _get_python_include_dirs()
1720 lpaths = cpp_extension.library_paths(cuda) + [
1774 ipaths = cpp_extension.include_paths(cuda) + _get_python_include_dirs()
1828 lpaths += [cpp_extension.TORCH_LIB_PATH]
3156 from torch.utils import cpp_extension
3158 lpaths = cpp_extension.library_paths(cuda=True) + [
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/
H A Dsetup.py7 from torch.utils.cpp_extension import BuildExtension, CppExtension
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dsetup.py8 from torch.utils.cpp_extension import (
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_custom_ops.py7 import torch.utils.cpp_extension

12