Home
last modified time | relevance | path

Searched refs:outer_loop_fusion_depth (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp.py329 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 …]