| /external/pytorch/torch/_inductor/codegen/rocm/ |
| D | rocm_cpp_scheduling.py | 44 def define_kernel(self, src_code: str, node_schedule) -> str: member in ROCmCPPScheduling
69 wrapper.define_kernel(
96 kernel_name = self.define_kernel(src_code, node_schedule)
|
| /external/pytorch/torch/_inductor/codegen/cuda/ |
| D | cuda_cpp_scheduling.py | 49 def define_kernel(self, src_code: str, node_schedule) -> str: member in CUDACPPScheduling
74 wrapper.define_kernel(
102 kernel_name = self.define_kernel(src_code, node_schedule)
|
| /external/autotest/metadata/ |
| D | tests.star | 17 load('//metadata/tests/kernel.star', define_kernel = 'define_tests')
70 tests += define_kernel()
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | multi_kernel.py | 88 def define_kernel(self, kernels): member in MultiKernelState
150 self.kernel_name = V.graph.wrapper_code.multi_kernel_state.define_kernel(
|
| D | cpp_wrapper_cuda.py | 197 def define_kernel( member in CppWrapperCuda
201 return super().define_kernel(name, kernel, metadata, cuda)
|
| D | simd.py | 1353 kernel_name2 = self.define_kernel(src_code2, node_schedule, kernel)
1366 kernel_name = self.define_kernel(src_code, node_schedule, kernel)
1499 kernel_name = self.define_kernel(src_code, node_schedule, kernel)
1598 kernel_name = self.define_kernel(src_code, [combo_kernel_node], kernel)
1821 def define_kernel(self, src_code, node_schedule, kernel): member in SIMDScheduling
|
| D | halide.py | 1672 def define_kernel(self, src_code, node_schedule, kernel): member in HalideScheduling
1693 wrapper.define_kernel(
|
| D | wrapper.py | 1251 def define_kernel( member in WrapperCodeGen
1463 self.define_kernel(
|
| D | cpp.py | 4530 kernel_name = self.define_kernel(src_code, node_schedule, kernel.args)
4544 def define_kernel(self, src_code, nodes, kernel_args=None): member in CppScheduling
4567 wrapper.define_kernel(kernel_name, compile_wrapper.getvalue(), cuda=False)
4573 kernel_name = self.define_kernel(
|
| D | triton.py | 3011 def define_kernel(self, src_code, node_schedule, kernel): member in TritonScheduling
3050 wrapper.define_kernel(
|
| D | cpp_wrapper_cpu.py | 923 def define_kernel( member in CppWrapperCpu
|
| /external/pytorch/torch/_inductor/ |
| D | utils.py | 1784 orig_define_kernel = WrapperCodeGen.define_kernel
|