File tree Expand file tree Collapse file tree 3 files changed +7
-5
lines changed
Expand file tree Collapse file tree 3 files changed +7
-5
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff line change @@ -102,8 +102,8 @@ list(
102102# Link CUDA backend
103103if (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 )
107107endif ()
108108
109109# Add tokenizers
Original file line number Diff line number Diff line change 6969# Link CUDA backend
7070if (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 )
7474endif ()
7575
7676if (EXECUTORCH_BUILD_METAL)
You can’t perform that action at this time.
0 commit comments