Home
last modified time | relevance | path

Searched refs:cpp_wrapper (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py303 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 Dcompile_fx.py444 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 Dir.py4085 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 Dconfig.py76 cpp_wrapper = os.environ.get("TORCHINDUCTOR_CPP_WRAPPER", "0") == "1" variable
H A Dselect_algorithm.py551 if V.graph.cpp_wrapper:
H A Dutils.py1332 self.codegen_with_cpp_wrapper() if self.cpp_wrapper else self.codegen()
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Ddebug_utils.py130 if V.graph.cpp_wrapper:
165 if V.graph.cpp_wrapper:
H A Dmulti_kernel.py111 if V.graph.cpp_wrapper:
173 if V.graph.cpp_wrapper:
H A Dwrapper.py304 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 Dcommon.py183 def get_wrapper_codegen_for_device(device: str, cpp_wrapper: bool = False):
188 if cpp_wrapper
H A Dcpp.py4552 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 Dtriton.py2846 if V.graph.cpp_wrapper:
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cuda_cpp_wrapper.py103 @config.patch(cpp_wrapper=True)
H A Dtest_cpu_cpp_wrapper.py134 @config.patch(cpp_wrapper=True)
H A Dtest_mkldnn_pattern_matcher.py1822 if torch._inductor.config.cpp_wrapper:
H A Dtest_torchinductor.py702 if config.cpp_wrapper:
1095 "view_dtype" if config.cpp_wrapper else "aten.view"
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py1783 cpp_wrapper: bool = False,