minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Revert "[Build] Enable full kernel in aarch64 wheel" (#18385)

Browse Source
This commit is contained in:
Baizhou Zhang
2026-02-07 09:19:07 +08:00
committed by GitHub
parent bedade1ef0
commit 9fbec79906
3 changed files with 1 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/release-whl-kernel.yml
-1
View File
@@ -47,7 +47,6 @@ jobs:
./build.sh "${{ matrix.python-version }}" "${{ matrix.cuda-version }}" ${{ matrix.arch == 'aarch64' && 'aarch64' || '' }} ./build.sh "${{ matrix.python-version }}" "${{ matrix.cuda-version }}" ${{ matrix.arch == 'aarch64' && 'aarch64' || '' }}
env: env:
USE_CCACHE: 0 USE_CCACHE: 0
CMAKE_EXTRA_ARGS: ${{ matrix.arch == 'aarch64' && '-DENABLE_BELOW_SM90=ON' || '' }}
- name: Upload to PyPI - name: Upload to PyPI
working-directory: sgl-kernel working-directory: sgl-kernel
sgl-kernel/Dockerfile
-5
View File
@@ -96,8 +96,6 @@ COPY . /sgl-kernel/
# Optional: enable CMake/Ninja profiling (pass non-empty via --build-arg ENABLE_*) # Optional: enable CMake/Ninja profiling (pass non-empty via --build-arg ENABLE_*)
ARG ENABLE_CMAKE_PROFILE ARG ENABLE_CMAKE_PROFILE
ARG ENABLE_BUILD_PROFILE ARG ENABLE_BUILD_PROFILE
# Optional: cmake extra args for aarch64 full kernel build (pass non-empty via --build-arg CMAKE_EXTRA_ARGS)
ARG CMAKE_EXTRA_ARGS
ARG ARCH=x86_64 ARG ARCH=x86_64
ARG USE_CCACHE=1 ARG USE_CCACHE=1
@@ -130,9 +128,6 @@ RUN --mount=type=cache,id=sgl-kernel-ccache,target=/ccache \
echo "CMake profiling enabled - will save to /sgl-kernel/cmake-profile.json"; \ echo "CMake profiling enabled - will save to /sgl-kernel/cmake-profile.json"; \
export CMAKE_ARGS="--profiling-output=/sgl-kernel/cmake-profile.json --profiling-format=google-trace"; \ export CMAKE_ARGS="--profiling-output=/sgl-kernel/cmake-profile.json --profiling-format=google-trace"; \
fi; \ fi; \
# Merge Extra CMake options into CMAKE_ARGS \
export CMAKE_ARGS="${CMAKE_ARGS:+${CMAKE_ARGS} }${CMAKE_EXTRA_ARGS:-}"; \
echo "CMAKE_ARGS= [$CMAKE_ARGS]"; \
${PYTHON_ROOT_PATH}/bin/python -m uv build --wheel -Cbuild-dir=build . --color=always --no-build-isolation; \ ${PYTHON_ROOT_PATH}/bin/python -m uv build --wheel -Cbuild-dir=build . --color=always --no-build-isolation; \
./rename_wheels.sh; \ ./rename_wheels.sh; \
if [ -n "${ENABLE_BUILD_PROFILE:-}" ] && [ -f /sgl-kernel/build/.ninja_log ]; then \ if [ -n "${ENABLE_BUILD_PROFILE:-}" ] && [ -f /sgl-kernel/build/.ninja_log ]; then \
sgl-kernel/build.sh
+1 -3
View File
@@ -48,12 +48,10 @@ echo "Buildx cache: ${BUILDX_CACHE_DIR}"
echo "Builder: ${BUILDER_NAME}" echo "Builder: ${BUILDER_NAME}"
echo "----------------------------------------" echo "----------------------------------------"
BUILD_ARGS=()
# Optional profiling build-args (empty string disables) # Optional profiling build-args (empty string disables)
BUILD_ARGS=()
[ -n "${ENABLE_CMAKE_PROFILE:-}" ] && BUILD_ARGS+=(--build-arg ENABLE_CMAKE_PROFILE="${ENABLE_CMAKE_PROFILE}") [ -n "${ENABLE_CMAKE_PROFILE:-}" ] && BUILD_ARGS+=(--build-arg ENABLE_CMAKE_PROFILE="${ENABLE_CMAKE_PROFILE}")
[ -n "${ENABLE_BUILD_PROFILE:-}" ] && BUILD_ARGS+=(--build-arg ENABLE_BUILD_PROFILE="${ENABLE_BUILD_PROFILE}") [ -n "${ENABLE_BUILD_PROFILE:-}" ] && BUILD_ARGS+=(--build-arg ENABLE_BUILD_PROFILE="${ENABLE_BUILD_PROFILE}")
# Optional extra cmake build-args (empty string disables)
[ -n "${CMAKE_EXTRA_ARGS:-}" ] && BUILD_ARGS+=(--build-arg CMAKE_EXTRA_ARGS="${CMAKE_EXTRA_ARGS}")
docker buildx build \ docker buildx build \
--builder "${BUILDER_NAME}" \ --builder "${BUILDER_NAME}" \