Skip to content

Commit d129e1e

Browse files
committed
solve ci
1 parent 4c60237 commit d129e1e

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
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)

examples/models/gemma3/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,8 @@ list(
102102
# Link CUDA backend
103103
if(EXECUTORCH_BUILD_CUDA)
104104
find_package(CUDAToolkit REQUIRED)
105-
list(APPEND link_libraries aoti_cuda)
106-
executorch_target_link_options_shared_lib(aoti_cuda)
105+
list(APPEND link_libraries aoti_cuda_backend)
106+
executorch_target_link_options_shared_lib(aoti_cuda_backend)
107107
endif()
108108

109109
# Add tokenizers

examples/models/whisper/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,8 @@ list(
6969
# Link CUDA backend
7070
if(EXECUTORCH_BUILD_CUDA)
7171
find_package(CUDAToolkit REQUIRED)
72-
list(APPEND _link_libraries aoti_cuda)
73-
executorch_target_link_options_shared_lib(aoti_cuda)
72+
list(APPEND _link_libraries aoti_cuda_backend)
73+
executorch_target_link_options_shared_lib(aoti_cuda_backend)
7474
endif()
7575

7676
if(EXECUTORCH_BUILD_METAL)

0 commit comments

Comments
 (0)