Searched refs:ExternKernelSchedulerNode (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | scheduler.py | 530 if isinstance(self, ExternKernelSchedulerNode) and isinstance(
646 if isinstance(self, ExternKernelSchedulerNode):
801 class ExternKernelSchedulerNode(BaseSchedulerNode): class
1481 extern = [x for x in nodes if isinstance(x, ExternKernelSchedulerNode)]
1491 if not isinstance(x, (NopKernelSchedulerNode, ExternKernelSchedulerNode))
1858 return ExternKernelSchedulerNode(self, node)
2174 ExternKernelSchedulerNode,
2897 isinstance(node1, (ExternKernelSchedulerNode, NopKernelSchedulerNode))
2903 isinstance(node2, (ExternKernelSchedulerNode, NopKernelSchedulerNode))
3318 def codegen_extern_call(self, scheduler_node: ExternKernelSchedulerNode) -> None: argument
[all …]
|
| H A D | comms.py | 527 isinstance(x, scheduler.ExternKernelSchedulerNode)
|