Skip to content

Commit 4be608d

Browse files
committed
solve ci
1 parent c9d871a commit 4be608d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

backends/aoti/aoti_backend.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,9 @@ def preprocess(
148148
options = cls.get_aoti_compile_options()
149149

150150
# Move the edge_program to the target device
151-
device_edge_program = move_to_device_pass(edge_program, device_name)
151+
device_edge_program = move_to_device_pass(
152+
edge_program, device_name if device_name != "metal" else "mps"
153+
)
152154

153155
# Replace view_copy with view
154156
ReplaceViewCopyWithViewPass()(device_edge_program.graph_module)

0 commit comments

Comments
 (0)