diff --git a/backends/apple/mps/partition/mps_partitioner.py b/backends/apple/mps/partition/mps_partitioner.py index 3dfc73cdd9..e5497389d1 100644 --- a/backends/apple/mps/partition/mps_partitioner.py +++ b/backends/apple/mps/partition/mps_partitioner.py @@ -102,7 +102,7 @@ def use_metal_kernel(self, node: torch.fx.Node): def tag_nodes(self, partitions: List[Partition]) -> None: for partition in partitions: crt_partition_counter = 0 - for node in sorted(partition.nodes): + for node in partition.nodes: delegation_tag = f"mps_{partition.id}" if self.use_metal_kernel(node): logging.warning(f"[WARNING] Using Metal kernel for op {node.name}!")