zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-20 01:11:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

set nccl ABI version (#2587)

Browse Source
This commit is contained in:
Awni Hannun
2025-09-11 16:55:53 -07:00
committed by GitHub
parent 6a3acf2301
commit af120c2bc0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/distributed/nccl/nccl_stub/CMakeLists.txt
View File

@@ -8,6 +8,7 @@ file(
"${CMAKE_CURRENT_BINARY_DIR}/nccl.h") "${CMAKE_CURRENT_BINARY_DIR}/nccl.h")
add_library(nccl SHARED nccl_stubs.cpp) add_library(nccl SHARED nccl_stubs.cpp)
set_target_properties(nccl PROPERTIES SOVERSION 2)
find_package(CUDAToolkit REQUIRED) find_package(CUDAToolkit REQUIRED)
target_include_directories(nccl PRIVATE ${CUDAToolkit_INCLUDE_DIRS}) target_include_directories(nccl PRIVATE ${CUDAToolkit_INCLUDE_DIRS})
target_include_directories(nccl PRIVATE ${CMAKE_CURRENT_BINARY_DIR}) target_include_directories(nccl PRIVATE ${CMAKE_CURRENT_BINARY_DIR})