File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -103,7 +103,7 @@ install(
103103)
104104
105105# CUDA-specific AOTI shim symbols (dynamically linked)
106- set (_aoti_cuda_shim_sources runtime/shims/memory.cpp
106+ set (_aoti_cuda_shim_sources runtime/cuda_allocator.cpp runtime/ shims/memory.cpp
107107 runtime/shims/cuda_guard.cpp
108108)
109109
@@ -180,8 +180,12 @@ install(
180180
181181# CUDA backend implementation
182182set (_aoti_cuda_backend_sources runtime/cuda_backend.cpp)
183+ if (_cuda_is_msvc_toolchain)
184+ # MSVC links aoti_cuda_backend into portable_lib without relying on C++
185+ # symbols exported from aoti_cuda_shims.dll.
186+ list (APPEND _aoti_cuda_backend_sources runtime/cuda_allocator.cpp)
187+ endif ()
183188
184- # CUDA backend implementation
185189add_library (aoti_cuda_backend STATIC ${_aoti_cuda_backend_sources} )
186190
187191target_include_directories (
You can’t perform that action at this time.
0 commit comments