Skip to content

Commit 9811eea

Browse files
DenisVieriu97facebook-github-bot
authored andcommitted
Remove the sorting of the nodes from partitioning (not needed for now as Custom Metal kernels are yet not enabled) (#3328)
Summary: Remove the sorting of the nodes from partitioning (not needed for now as Custom Metal kernels are yet not enabled) **Testing**: Verified that tracing works correctly with release branch: `python3 -m examples.apple.mps.scripts.mps_example --model_name="mv3"` cc shoumikhin , cccclai Pull Request resolved: #3328 Reviewed By: shoumikhin Differential Revision: D56540389 Pulled By: cccclai fbshipit-source-id: e8a53f624b58ac4d2348c87e08acd5f2fb3de5b2
1 parent 453ebad commit 9811eea

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)