Searched defs:node_schedule (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | simd.py | 1105 def requires_closing_previous_reduction(node, node_schedule): argument
1206 def select_index_dtype(cls, node_schedule, numel, reduction_numel): argument
1235 def has_non_contiguous_pw_in_reduction_kernel(self, node_schedule, numel, rnumel): argument
1258 def get_kernel_args(self, node_schedule, numel, reduction_numel): argument
1295 def codegen_node_schedule( argument
1412 def codegen_node_schedule_with_kernel(self, node_schedule, kernel): argument
1675 def select_tiling(cls, node_schedule, numel, reduction_numel=sympy.Integer(1)): argument
1818 def codegen_comment(self, node_schedule): argument
1821 def define_kernel(self, src_code, node_schedule, kernel): argument
1852 def filter(node_schedule): argument
|
| H A D | triton.py | 2985 def codegen_comment(self, node_schedule): argument
3011 def define_kernel(self, src_code, node_schedule, kernel): argument
|
| H A D | halide.py | 1672 def define_kernel(self, src_code, node_schedule, kernel): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | utils.py | 483 def aggregate_origins(node_schedule): argument
502 def get_fused_kernel_name(node_schedule, descriptive_names): argument
535 def get_kernel_metadata(node_schedule, wrapper): argument
1853 def dump_node_schedule(node_schedule): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ |
| H A D | rocm_cpp_scheduling.py | 44 def define_kernel(self, src_code: str, node_schedule) -> str: argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cuda_cpp_scheduling.py | 49 def define_kernel(self, src_code: str, node_schedule) -> str: argument
|