ref(upstream): FULL TREE — Deep-Spark xllm (1470) + ds_vllm csrc/models (703)
Replaces cherry-picked upstream_ref with complete source trees. xllm/ — Iluvatar official C++ inference engine (15MB, 1470 files) Complete: kernels → layers → models → runtime → scheduler → api Excluded: .git, binary images, third_party submodule checkouts ds_vllm/ — Iluvatar official vllm fork (8MB, 703 files) Included: csrc/ (ALL CUDA kernels), fused_moe/, qwen3_5 model, _custom_ops Excluded: tests, benchmarks, docs, examples (not needed for reference) Critical call chains now fully traceable: MoE: moe_topk_softmax_kernels.cuh → ixformer.h → fused_moe.cpp → layer GDN: qwen3_gated_delta_net_base.cpp → qwen3_5_gated_delta_net.cpp Attention: ixformer.h → xllm_paged_attention → attention.cpp
This commit is contained in:
93
upstream_ref/xllm/cmake/cargo_shared_library.cmake
Normal file
93
upstream_ref/xllm/cmake/cargo_shared_library.cmake
Normal file
@@ -0,0 +1,93 @@
|
||||
include(CMakeParseArguments)
|
||||
|
||||
# inspired by https://github.com/abseil/abseil-cpp
|
||||
# cc_library()
|
||||
# CMake function to imitate Bazel's cc_library rule.
|
||||
function(cargo_shared_library)
|
||||
cmake_parse_arguments(
|
||||
CARGO # prefix
|
||||
"" # options
|
||||
"NAME" # one value args
|
||||
"HDRS" # multi value args
|
||||
${ARGN}
|
||||
)
|
||||
|
||||
string(REPLACE "-" "_" LIB_NAME ${CARGO_NAME})
|
||||
# set(CARGO_TARGET_DIR ${CMAKE_CURRENT_BINARY_DIR})
|
||||
|
||||
# figure out the target triple
|
||||
if(WIN32)
|
||||
if(CMAKE_SIZEOF_VOID_P EQUAL 8)
|
||||
set(LIB_TARGET "x86_64-pc-windows-msvc")
|
||||
else()
|
||||
set(LIB_TARGET "i686-pc-windows-msvc")
|
||||
endif()
|
||||
elseif(ANDROID)
|
||||
if(ANDROID_SYSROOT_ABI STREQUAL "x86")
|
||||
set(LIB_TARGET "i686-linux-android")
|
||||
elseif(ANDROID_SYSROOT_ABI STREQUAL "x86_64")
|
||||
set(LIB_TARGET "x86_64-linux-android")
|
||||
elseif(ANDROID_SYSROOT_ABI STREQUAL "arm")
|
||||
set(LIB_TARGET "arm-linux-androideabi")
|
||||
elseif(ANDROID_SYSROOT_ABI STREQUAL "arm64")
|
||||
set(LIB_TARGET "aarch64-linux-android")
|
||||
endif()
|
||||
elseif(IOS)
|
||||
set(LIB_TARGET "universal")
|
||||
elseif(CMAKE_SYSTEM_NAME STREQUAL Darwin)
|
||||
set(LIB_TARGET "x86_64-apple-darwin")
|
||||
else()
|
||||
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
|
||||
set(LIB_TARGET "aarch64-unknown-linux-gnu")
|
||||
elseif(CMAKE_SIZEOF_VOID_P EQUAL 8)
|
||||
set(LIB_TARGET "x86_64-unknown-linux-gnu")
|
||||
else()
|
||||
set(LIB_TARGET "i686-unknown-linux-gnu")
|
||||
endif()
|
||||
endif()
|
||||
|
||||
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
||||
set(LIB_BUILD_TYPE "debug")
|
||||
else()
|
||||
set(LIB_BUILD_TYPE "release")
|
||||
endif()
|
||||
|
||||
if(IOS)
|
||||
set(CARGO_ARGS "lipo")
|
||||
else()
|
||||
set(CARGO_ARGS "build")
|
||||
list(APPEND CARGO_ARGS "--target" ${LIB_TARGET})
|
||||
endif()
|
||||
|
||||
if(${LIB_BUILD_TYPE} STREQUAL "release")
|
||||
list(APPEND CARGO_ARGS "--release")
|
||||
endif()
|
||||
|
||||
file(GLOB_RECURSE LIB_SOURCES "*.rs")
|
||||
|
||||
set(CARGO_ENV_COMMAND ${CMAKE_COMMAND} -E env "CARGO_TARGET_DIR=${CMAKE_CURRENT_BINARY_DIR}")
|
||||
|
||||
# build the library target with cargo
|
||||
set(SHARED_LIB_NAME
|
||||
"${CMAKE_SHARED_LIBRARY_PREFIX}${LIB_NAME}${CMAKE_SHARED_LIBRARY_SUFFIX}")
|
||||
set(LIB_FILE
|
||||
"${CMAKE_CURRENT_BINARY_DIR}/${LIB_TARGET}/${LIB_BUILD_TYPE}/${SHARED_LIB_NAME}")
|
||||
|
||||
add_custom_command(
|
||||
OUTPUT ${LIB_FILE}
|
||||
COMMAND ${CARGO_ENV_COMMAND} ${CARGO_EXECUTABLE} ARGS ${CARGO_ARGS}
|
||||
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
DEPENDS ${LIB_SOURCES}
|
||||
COMMENT "Building cargo library ${LIB_FILE}"
|
||||
)
|
||||
add_custom_target(${CARGO_NAME}_target ALL DEPENDS ${LIB_FILE})
|
||||
|
||||
# add the library target
|
||||
add_library(${CARGO_NAME} SHARED IMPORTED GLOBAL)
|
||||
add_dependencies(${CARGO_NAME} ${CARGO_NAME}_target)
|
||||
set_target_properties(${CARGO_NAME} PROPERTIES
|
||||
IMPORTED_LOCATION ${LIB_FILE}
|
||||
INTERFACE_INCLUDE_DIRECTORIES ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
IMPORTED_NO_SONAME TRUE
|
||||
)
|
||||
endfunction()
|
||||
Reference in New Issue
Block a user