Searched defs:FusedSchedulerNode (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cuda_combined_scheduling.py | 79 def codegen_node(self, node: Union[FusedSchedulerNode, SchedulerNode]):
|
| H A D | simd.py | 1137 self, node: Union[scheduler.FusedSchedulerNode, scheduler.SchedulerNode]
|
| H A D | cpp.py | 366 outer_fused_nodes: List[Union[FusedSchedulerNode, SchedulerNode]],
4456 node: Union[OuterLoopFusedSchedulerNode, FusedSchedulerNode, SchedulerNode], argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | scheduler.py | 1076 class FusedSchedulerNode(BaseSchedulerNode): class
3646 def codegen_node(self, node: Union[FusedSchedulerNode, SchedulerNode]) -> None:
3699 def debug_triton_code(node: Union[SchedulerNode, FusedSchedulerNode]) -> List[str]: argument
|
| H A D | debug.py | 150 def in_output(snode: Union[BaseSchedulerNode, FusedSchedulerNode]) -> bool: argument
|