From 9fbec799063b23f9d75a7d81006d0612b8bbd547 Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <sobereddiezhang@gmail.com>
Date: Sat, 7 Feb 2026 09:19:07 +0800
Subject: [PATCH] Revert "[Build] Enable full kernel in aarch64 wheel" (#18385)

---
 .github/workflows/release-whl-kernel.yml | 1 -
 sgl-kernel/Dockerfile                    | 5 -----
 sgl-kernel/build.sh                      | 4 +---
 3 files changed, 1 insertion(+), 9 deletions(-)

diff --git a/.github/workflows/release-whl-kernel.yml b/.github/workflows/release-whl-kernel.yml
index 2fe1e8aef..1c4b29397 100644
--- a/.github/workflows/release-whl-kernel.yml
+++ b/.github/workflows/release-whl-kernel.yml
@@ -47,7 +47,6 @@ jobs:
           ./build.sh "${{ matrix.python-version }}" "${{ matrix.cuda-version }}" ${{ matrix.arch == 'aarch64' && 'aarch64' || '' }}
         env:
           USE_CCACHE: 0
-          CMAKE_EXTRA_ARGS: ${{ matrix.arch == 'aarch64' && '-DENABLE_BELOW_SM90=ON' || '' }}
 
       - name: Upload to PyPI
         working-directory: sgl-kernel
diff --git a/sgl-kernel/Dockerfile b/sgl-kernel/Dockerfile
index 08063674a..e5bc1c339 100644
--- a/sgl-kernel/Dockerfile
+++ b/sgl-kernel/Dockerfile
@@ -96,8 +96,6 @@ COPY . /sgl-kernel/
 # Optional: enable CMake/Ninja profiling (pass non-empty via --build-arg ENABLE_*)
 ARG ENABLE_CMAKE_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 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"; \
       export CMAKE_ARGS="--profiling-output=/sgl-kernel/cmake-profile.json --profiling-format=google-trace"; \
     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; \
     ./rename_wheels.sh; \
     if [ -n "${ENABLE_BUILD_PROFILE:-}" ] && [ -f /sgl-kernel/build/.ninja_log ]; then \
diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh
index ea9be7df5..b29d4d64a 100755
--- a/sgl-kernel/build.sh
+++ b/sgl-kernel/build.sh
@@ -48,12 +48,10 @@ echo "Buildx cache:   ${BUILDX_CACHE_DIR}"
 echo "Builder:        ${BUILDER_NAME}"
 echo "----------------------------------------"
 
-BUILD_ARGS=()
 # 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_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 \
   --builder "${BUILDER_NAME}" \