Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dscheduler.py530 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 Dcomms.py527 isinstance(x, scheduler.ExternKernelSchedulerNode)