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)
|