add target argument for cross compiling

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Sep 30, 2023 · e0c6f4f · e0c6f4f
1 parent 4c6cf7e
commit e0c6f4f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/lib/src/gpu/cudart/CMakeLists.txt b/source/lib/src/gpu/cudart/CMakeLists.txt
@@ -12,7 +12,8 @@ execute_process(
   COMMAND
     ${Python_EXECUTABLE}
     ${CMAKE_CURRENT_SOURCE_DIR}/../../../../3rdparty/implib/implib-gen.py
-    ${CUDART_LOCATION} COMMAND_ERROR_IS_FATAL ANY
+    ${CUDART_LOCATION} --target ${CMAKE_SYSTEM_PROCESSOR} COMMAND_ERROR_IS_FATAL
+    ANY
   WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
 
 file(GLOB CUDA_STUB_SRC ${CMAKE_CURRENT_BINARY_DIR}/*.tramp.S