Skip to content

Commit c70b27b

Browse files
committed
Remove the sorting of the nodes from partitioning (not needed for now as Custom Metal kernels are yet not enabled)
1 parent 035aee4 commit c70b27b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/apple/mps/partition/mps_partitioner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def use_metal_kernel(self, node: torch.fx.Node):
102102
def tag_nodes(self, partitions: List[Partition]) -> None:
103103
for partition in partitions:
104104
crt_partition_counter = 0
105-
for node in sorted(partition.nodes):
105+
for node in partition.nodes:
106106
delegation_tag = f"mps_{partition.id}"
107107
if self.use_metal_kernel(node):
108108
logging.warning(f"[WARNING] Using Metal kernel for op {node.name}!")

0 commit comments

Comments
 (0)