Searched refs:ForeachKernelSchedulerNode (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| 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 …]
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | simd.py | 957 if isinstance(node1, scheduler.ForeachKernelSchedulerNode) or isinstance(
958 node2, scheduler.ForeachKernelSchedulerNode
960 return scheduler.ForeachKernelSchedulerNode.can_fuse(node1, node2)
|
| D | triton.py | 2994 ForeachKernelSchedulerNode,
2998 isinstance(n, ForeachKernelSchedulerNode) for n in node_schedule
|
| D | cpp.py | 29 ForeachKernelSchedulerNode,
3929 return ForeachKernelSchedulerNode.fuse(node1, node2)
|