We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c9d871a commit 4be608dCopy full SHA for 4be608d
backends/aoti/aoti_backend.py
@@ -148,7 +148,9 @@ def preprocess(
148
options = cls.get_aoti_compile_options()
149
150
# Move the edge_program to the target device
151
- device_edge_program = move_to_device_pass(edge_program, device_name)
+ device_edge_program = move_to_device_pass(
152
+ edge_program, device_name if device_name != "metal" else "mps"
153
+ )
154
155
# Replace view_copy with view
156
ReplaceViewCopyWithViewPass()(device_edge_program.graph_module)
0 commit comments