Searched refs:outer_loop_fusion_depth (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
H A D | cpp.py | 329 cls, node1: BaseSchedulerNode, node2: BaseSchedulerNode, outer_loop_fusion_depth argument 358 outer_loop_fusion_depth, 361 … return cls(node1.scheduler, [node1, node2], outer_loop_fusion_depth) # type: ignore[list-item] 367 outer_loop_fusion_depth, argument 372 self.outer_loop_fusion_depth = outer_loop_fusion_depth 383 self, cpp_kernel_proxy_list, outer_loop_fusion_depth argument 442 left_loop_nest.root[0], right_loop_nest.root[0], outer_loop_fusion_depth 459 outer_loop_fusion_depth, argument 461 assert outer_loop_fusion_depth >= 1 466 if (outer_loop_fusion_depth := outer_loop_fusion_depth - 1) >= 1: [all …]
|