Searched refs:extern_kernel (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp_wrapper_cpu.py | 1259 def generate_c_shim_extern_kernel_alloc(self, extern_kernel, args): argument
1261 name = extern_kernel.name
1266 extern_kernel.get_kernel_name(), args + [output_arg]
1270 def generate_extern_kernel_alloc(self, extern_kernel, args): argument
1272 self.generate_c_shim_extern_kernel_alloc(extern_kernel, args)
1274 super().generate_extern_kernel_alloc(extern_kernel, args)
|
| D | wrapper.py | 755 def generate_extern_kernel_alloc(self, extern_kernel, args): argument
758 no_return = isinstance(extern_kernel.layout, ir.NoneLayout)
759 output_name = extern_kernel.get_name()
760 origin_node = extern_kernel.get_origin_node()
761 kernel_name = extern_kernel.get_kernel_name()
|