Searched refs:cpp_wrapper (Results 1 – 17 of 17) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | graph.py | 303 cpp_wrapper=False, argument 385 self.cpp_wrapper = cpp_wrapper 390 self.record_multi_kernel_choice = cpp_wrapper 616 cpp_wrapper=self.cpp_wrapper, 1516 if self.cpp_wrapper: 1531 device_type, self.cpp_wrapper 1553 self.cpp_wrapper = False 1625 self.cpp_wrapper = True 1663 self.cpp_wrapper = parent_graph.cpp_wrapper 1684 self.codegen_with_cpp_wrapper() if self.cpp_wrapper else self.codegen() [all …]
|
H A D | compile_fx.py | 444 cpp_wrapper: bool = False, 481 cpp_wrapper=cpp_wrapper, 500 "cpp_wrapper": cpp_wrapper, 694 cpp_wrapper: bool = False, 788 cpp_wrapper=cpp_wrapper, 796 assert cpp_wrapper, "AOT mode only supports C++ wrapper" 809 cpp_wrapper=cpp_wrapper, 1277 if config.cpp_wrapper: 1306 inner_compile=functools.partial(inner_compile, cpp_wrapper=True),
|
H A D | ir.py | 4085 if V.graph.cpp_wrapper 4391 if V.graph.cpp_wrapper: 4415 if V.graph.cpp_wrapper: 4439 if V.graph.cpp_wrapper: 4468 if config.size_asserts and not V.graph.cpp_wrapper: 4670 if V.graph.cpp_wrapper: 4807 if V.graph.cpp_wrapper and config.abi_compatible: 4971 if V.graph.cpp_wrapper: 5173 if V.graph.cpp_wrapper: 5370 if V.graph.cpp_wrapper and config.abi_compatible: [all …]
|
H A D | config.py | 76 cpp_wrapper = os.environ.get("TORCHINDUCTOR_CPP_WRAPPER", "0") == "1" variable
|
H A D | select_algorithm.py | 551 if V.graph.cpp_wrapper:
|
H A D | utils.py | 1332 self.codegen_with_cpp_wrapper() if self.cpp_wrapper else self.codegen()
|
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
H A D | debug_utils.py | 130 if V.graph.cpp_wrapper: 165 if V.graph.cpp_wrapper:
|
H A D | multi_kernel.py | 111 if V.graph.cpp_wrapper: 173 if V.graph.cpp_wrapper:
|
H A D | wrapper.py | 304 if V.graph.cpp_wrapper: 347 if not V.graph.cpp_wrapper: 868 if V.graph.aot_mode and V.graph.cpp_wrapper and V.graph.is_const_graph:
|
H A D | common.py | 183 def get_wrapper_codegen_for_device(device: str, cpp_wrapper: bool = False): 188 if cpp_wrapper
|
H A D | cpp.py | 4552 kernel_decl_name = kernel_name if V.graph.cpp_wrapper else "kernel" 4562 if not V.graph.cpp_wrapper: 4565 if not V.graph.cpp_wrapper:
|
H A D | triton.py | 2846 if V.graph.cpp_wrapper:
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_cuda_cpp_wrapper.py | 103 @config.patch(cpp_wrapper=True)
|
H A D | test_cpu_cpp_wrapper.py | 134 @config.patch(cpp_wrapper=True)
|
H A D | test_mkldnn_pattern_matcher.py | 1822 if torch._inductor.config.cpp_wrapper:
|
H A D | test_torchinductor.py | 702 if config.cpp_wrapper: 1095 "view_dtype" if config.cpp_wrapper else "aten.view"
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_subclasses.py | 1783 cpp_wrapper: bool = False,
|