Home
last modified time | relevance | path

Searched refs:ForeachKernelSchedulerNode (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dscheduler.py1265 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 Dsimd.py957 if isinstance(node1, scheduler.ForeachKernelSchedulerNode) or isinstance(
958 node2, scheduler.ForeachKernelSchedulerNode
960 return scheduler.ForeachKernelSchedulerNode.can_fuse(node1, node2)
H A Dtriton.py2994 ForeachKernelSchedulerNode,
2998 isinstance(n, ForeachKernelSchedulerNode) for n in node_schedule
H A Dcpp.py29 ForeachKernelSchedulerNode,
3929 return ForeachKernelSchedulerNode.fuse(node1, node2)