Skip to content

Commit

Permalink
Remove the sorting of the nodes from partitioning (not needed for now…
Browse files Browse the repository at this point in the history
… as Custom Metal kernels are yet not enabled) (#3327)
  • Loading branch information
DenisVieriu97 authored Apr 24, 2024
1 parent c78c45b commit a1d881a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion backends/apple/mps/partition/mps_partitioner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}!")
Expand Down

0 comments on commit a1d881a

Please sign in to comment.