/aosp_15_r20/external/pytorch/test/ |
H A D | test_cpp_extensions_jit.py | 17 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 D | test_cpp_extensions_aot.py | 12 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 D | test_cpp_extensions_stream_and_event.py | 11 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 D | test_cpp_extensions_mtia_backend.py | 11 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 D | test_cpp_extensions_open_device_registration.py | 17 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 D | test_utils.py | 20 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 D | test_dispatch.py | 9 import torch.utils.cpp_extension 962 module = torch.utils.cpp_extension.load(
|
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ |
H A D | compile_command.py | 14 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 D | verify_dynamo.py | 36 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 D | cpp_jit.py | 13 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 D | test_extension_backend.py | 9 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 D | test_compiled_autograd.py | 1591 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 D | test_triton_extension_backend.py | 9 import torch.utils.cpp_extension
|
/aosp_15_r20/external/pytorch/test/profiler/ |
H A D | test_cpp_thread.py | 9 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 D | cpp_extension.rst | 1 torch.utils.cpp_extension 4 .. currentmodule:: torch.utils.cpp_extension
|
H A D | index.rst | 120 torch.utils.cpp_extension <cpp_extension>
|
/aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
H A D | utils.py | 9 import torch.utils.cpp_extension 131 cpp_module = torch.utils.cpp_extension.load_inline(
|
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/ |
H A D | benchmark_core.py | 15 import torch.utils.cpp_extension as cpp_extension # noqa: F401 namespace
|
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt_extension/ |
H A D | setup.py | 3 from torch.utils.cpp_extension import BuildExtension, CppExtension
|
/aosp_15_r20/external/pytorch/test/cpp_extensions/no_python_abi_suffix_test/ |
H A D | setup.py | 3 from torch.utils.cpp_extension import BuildExtension, CppExtension
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | cpp_builder.py | 746 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 D | codecache.py | 1699 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 D | setup.py | 7 from torch.utils.cpp_extension import BuildExtension, CppExtension
|
/aosp_15_r20/external/pytorch/test/cpp_extensions/ |
H A D | setup.py | 8 from torch.utils.cpp_extension import (
|
/aosp_15_r20/external/pytorch/test/onnx/ |
H A D | test_custom_ops.py | 7 import torch.utils.cpp_extension
|