mirror of
https://github.com/deepseek-ai/DeepEP
synced 2025-06-26 18:28:11 +00:00
b8d90fb753
* Update README * Update `setup.py` * Fix headers * Add `DISABLE_NVSHMEM` for APIs * Fix launch * Fix TMA settings * Fix TMA usages * Fix dlink * Separate layout kernels * Update version * Add `is_sm90_compiled` * Fix tests * Add NVLink connection checks * Update README * Fix tests * Add some comments * Minor fix * Minor fix * Fix bugs
22 lines
887 B
CMake
22 lines
887 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(runtime_cuda runtime.cu)
|
|
add_deep_ep_library(layout_cuda layout.cu)
|
|
add_deep_ep_library(intranode_cuda intranode.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 runtime_cuda layout_cuda intranode_cuda internode_cuda internode_ll_cuda PARENT_SCOPE)
|