Searched refs:ForeachKernelSchedulerNode (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | scheduler.py | 1265 class ForeachKernelSchedulerNode(FusedSchedulerNode): class 1303 producer = typing.cast(ForeachKernelSchedulerNode, producer) 1304 consumer = typing.cast(ForeachKernelSchedulerNode, consumer) 1319 consumer = typing.cast(ForeachKernelSchedulerNode, consumer) 1334 producer = typing.cast(ForeachKernelSchedulerNode, producer) 1349 ) -> ForeachKernelSchedulerNode: 1352 producer = typing.cast(ForeachKernelSchedulerNode, producer) 1356 consumer = typing.cast(ForeachKernelSchedulerNode, consumer) 1363 producer = typing.cast(ForeachKernelSchedulerNode, producer) 1364 consumer = typing.cast(ForeachKernelSchedulerNode, consumer) [all …]
|
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
H A D | simd.py | 957 if isinstance(node1, scheduler.ForeachKernelSchedulerNode) or isinstance( 958 node2, scheduler.ForeachKernelSchedulerNode 960 return scheduler.ForeachKernelSchedulerNode.can_fuse(node1, node2)
|
H A D | triton.py | 2994 ForeachKernelSchedulerNode, 2998 isinstance(n, ForeachKernelSchedulerNode) for n in node_schedule
|
H A D | cpp.py | 29 ForeachKernelSchedulerNode, 3929 return ForeachKernelSchedulerNode.fuse(node1, node2)
|