Searched refs:fallback_kernel (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cpu.py | 1276 def generate_c_shim_fallback_kernel(self, fallback_kernel, args): argument
1279 output_name_base = fallback_kernel.get_name()
1280 for idx, output in enumerate(fallback_kernel.outputs):
1307 self.generate_c_shim_extern_kernel_call(fallback_kernel.cpp_kernel_name, args)
1311 def generate_fallback_kernel(self, fallback_kernel, args): argument
1313 self.generate_c_shim_fallback_kernel(fallback_kernel, args)
1315 super().generate_fallback_kernel(fallback_kernel, args)
|
| H A D | wrapper.py | 752 def generate_fallback_kernel(self, fallback_kernel, args): argument
753 self.generate_extern_kernel_alloc(fallback_kernel, args)
|