Skip to content

Commit 0e8e201

Browse files
committed
fix(iluvatar): avoid unavailable CUDA driver target
1 parent ef4045a commit 0e8e201

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

‎examples/CMakeLists.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ foreach(source_file ${EXAMPLE_SOURCES})
3232
RULE_LAUNCH_COMPILE "${ILUVATAR_CUDA_COMPILER} "
3333
)
3434
target_compile_options(${target_name} PRIVATE ${ILUVATAR_CUDA_FLAGS} "-Wno-unused-command-line-argument")
35-
target_link_libraries(${target_name} PRIVATE CUDA::cudart CUDA::cuda_driver)
35+
target_link_libraries(${target_name} PRIVATE CUDA::cudart)
3636
endif()
3737

3838
if(WITH_METAX)

‎src/CMakeLists.txt‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,9 @@ if(WITH_ILUVATAR)
7676
set(CMAKE_CXX_COMPILER "${ILUVATAR_CUDA_COMPILER}" CACHE FILEPATH "Iluvatar Compiler" FORCE)
7777
target_compile_options(infiniccl PRIVATE ${ILUVATAR_CUDA_FLAGS})
7878

79-
target_link_libraries(infiniccl PRIVATE CUDA::cudart CUDA::cuda_driver)
79+
# CoreX does not export CUDA::cuda_driver; this backend only uses the
80+
# CUDA runtime API and NCCL.
81+
target_link_libraries(infiniccl PRIVATE CUDA::cudart)
8082
endif()
8183

8284
# MetaX

0 commit comments

Comments
 (0)