chore: bump sgl-kernel version to 0.4.2.post1 (#24457)
Co-authored-by: sglang-bot <sglang-bot@users.noreply.github.com> Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
This commit is contained in:
co-authored by
sglang-bot
Baizhou Zhang
parent
08d4c2072b
commit
6764155914
@@ -52,15 +52,6 @@ FetchContent_Declare(
|
||||
)
|
||||
FetchContent_Populate(repo-cutlass)
|
||||
|
||||
# DeepGEMM
|
||||
FetchContent_Declare(
|
||||
repo-deepgemm
|
||||
GIT_REPOSITORY https://github.com/sgl-project/DeepGEMM
|
||||
GIT_TAG ffe2b6b97420a9f8c58268ca55755168e6e2f360
|
||||
GIT_SHALLOW OFF
|
||||
)
|
||||
FetchContent_Populate(repo-deepgemm)
|
||||
|
||||
# fmt
|
||||
FetchContent_Declare(
|
||||
repo-fmt
|
||||
@@ -515,51 +506,6 @@ install(TARGETS spatial_ops LIBRARY DESTINATION sgl_kernel)
|
||||
# ============================ Extra Install: FLashMLA ============================= #
|
||||
include(${CMAKE_CURRENT_LIST_DIR}/cmake/flashmla.cmake)
|
||||
|
||||
# ============================ Extra Install: DeepGEMM (JIT) ============================= #
|
||||
# Create a separate library for DeepGEMM's Python API.
|
||||
# This keeps its compilation isolated from the main common_ops.
|
||||
set(DEEPGEMM_SOURCES
|
||||
"${repo-deepgemm_SOURCE_DIR}/csrc/python_api.cpp"
|
||||
)
|
||||
|
||||
Python_add_library(deep_gemm_cpp MODULE USE_SABI ${SKBUILD_SABI_VERSION} WITH_SOABI ${DEEPGEMM_SOURCES})
|
||||
|
||||
# Link against necessary libraries, including nvrtc for JIT compilation.
|
||||
target_link_libraries(deep_gemm_cpp PRIVATE ${TORCH_LIBRARIES} c10 cuda nvrtc mscclpp_static)
|
||||
|
||||
# Add include directories needed by DeepGEMM.
|
||||
target_include_directories(deep_gemm_cpp PRIVATE
|
||||
${repo-deepgemm_SOURCE_DIR}/deep_gemm/include
|
||||
${repo-cutlass_SOURCE_DIR}/include
|
||||
${repo-fmt_SOURCE_DIR}/include
|
||||
)
|
||||
|
||||
# Apply the same compile options as common_ops.
|
||||
target_compile_options(deep_gemm_cpp PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:${SGL_KERNEL_CUDA_FLAGS}>)
|
||||
|
||||
# Create an empty __init__.py to make `deepgemm` a Python package.
|
||||
file(WRITE ${CMAKE_CURRENT_BINARY_DIR}/deepgemm_pkg_init.py "")
|
||||
install(
|
||||
FILES ${CMAKE_CURRENT_BINARY_DIR}/deepgemm_pkg_init.py
|
||||
DESTINATION deep_gemm
|
||||
RENAME __init__.py
|
||||
)
|
||||
|
||||
# Install the compiled DeepGEMM API library.
|
||||
install(TARGETS deep_gemm_cpp LIBRARY DESTINATION deep_gemm)
|
||||
|
||||
# Install the source files required by DeepGEMM for runtime JIT compilation.
|
||||
install(
|
||||
DIRECTORY ${repo-deepgemm_SOURCE_DIR}/deep_gemm/
|
||||
DESTINATION deep_gemm
|
||||
)
|
||||
|
||||
install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cute/"
|
||||
DESTINATION "deep_gemm/include/cute")
|
||||
|
||||
install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cutlass/"
|
||||
DESTINATION "deep_gemm/include/cutlass")
|
||||
|
||||
# ============================ Extra Install: triton kernels ============================= #
|
||||
install(DIRECTORY "${repo-triton_SOURCE_DIR}/python/triton_kernels/triton_kernels/"
|
||||
DESTINATION "triton_kernels"
|
||||
|
||||
@@ -8,7 +8,7 @@ build-backend = "scikit_build_core.build"
|
||||
|
||||
[project]
|
||||
name = "sglang-kernel"
|
||||
version = "0.4.2"
|
||||
version = "0.4.2.post1"
|
||||
authors = [
|
||||
{ name="SGLang Kernel Team", email="sglang@lmsys.org" },
|
||||
]
|
||||
|
||||
@@ -8,7 +8,7 @@ build-backend = "scikit_build_core.build"
|
||||
|
||||
[project]
|
||||
name = "sglang-kernel-cpu"
|
||||
version = "0.4.2"
|
||||
version = "0.4.2.post1"
|
||||
description = "Kernel Library for SGLang"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.10"
|
||||
|
||||
@@ -10,7 +10,7 @@ build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
name = "sglang-kernel"
|
||||
version = "0.4.2"
|
||||
version = "0.4.2.post1"
|
||||
description = "Kernel Library for SGLang"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.10"
|
||||
|
||||
@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
name = "sglang-kernel"
|
||||
version = "0.4.2"
|
||||
version = "0.4.2.post1"
|
||||
description = "Kernel Library for SGLang"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.10"
|
||||
|
||||
@@ -1 +1 @@
|
||||
__version__ = "0.4.2"
|
||||
__version__ = "0.4.2.post1"
|
||||
|
||||
Reference in New Issue
Block a user