mirror of
https://github.com/deepseek-ai/DeepEP
synced 2025-05-01 18:58:25 +00:00
21 lines
832 B
CMake
21 lines
832 B
CMake
function(add_deep_ep_library target_name source_file)
|
|
add_library(${target_name} STATIC ${source_file})
|
|
set_target_properties(${target_name} PROPERTIES
|
|
POSITION_INDEPENDENT_CODE ON
|
|
CXX_STANDARD_REQUIRED ON
|
|
CUDA_STANDARD_REQUIRED ON
|
|
CXX_STANDARD 14
|
|
CUDA_STANDARD 14
|
|
CUDA_SEPARABLE_COMPILATION ON
|
|
)
|
|
target_link_libraries(${target_name} PUBLIC nvshmem cudart cudadevrt mlx5)
|
|
endfunction()
|
|
|
|
add_deep_ep_library(intranode_cuda intranode.cu)
|
|
add_deep_ep_library(runtime_cuda runtime.cu)
|
|
add_deep_ep_library(internode_cuda internode.cu)
|
|
add_deep_ep_library(internode_ll_cuda internode_ll.cu)
|
|
|
|
# Later, we should link all libraries in `EP_CUDA_LIBRARIES`
|
|
set(EP_CUDA_LIBRARIES intranode_cuda runtime_cuda internode_cuda internode_ll_cuda PARENT_SCOPE)
|