[Kernel] Move sgl-kernel under sglang.kernels.aot (#32648)

scripts/ci/amd/amd_ci_install_dependency.sh

@@ -128,7 +128,7 @@ else
   # Also clear cache in sglang-checkout
   docker exec ci_sglang find /sglang-checkout -name "*.pyc" -delete || true
   docker exec ci_sglang find /sglang-checkout -name "__pycache__" -type d -exec rm -rf {} + || true
-  docker exec -w /sglang-checkout/sgl-kernel ci_sglang bash -c "rm -f pyproject.toml && mv pyproject_rocm.toml pyproject.toml && python3 setup_rocm.py install"
+  docker exec -w /sglang-checkout/python/sglang/kernels/aot ci_sglang bash -c "rm -f pyproject.toml && mv pyproject_rocm.toml pyproject.toml && python3 setup_rocm.py install"
 
   docker exec ci_sglang bash -c 'rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml'
   install_with_retry docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache -e "python[${EXTRAS}]"

scripts/ci/cuda/ci_install_dependency.sh

@@ -304,28 +304,28 @@ install_sglang() {
 }
 
 install_sglang_kernel() {
-    SGL_KERNEL_VERSION_FROM_KERNEL=$(grep -Po '(?<=^version = ")[^"]*' sgl-kernel/pyproject.toml)
+    SGL_KERNEL_VERSION_FROM_KERNEL=$(grep -Po '(?<=^version = ")[^"]*' python/sglang/kernels/aot/pyproject.toml)
     SGL_KERNEL_VERSION_FROM_SRT=$(grep -Po -m1 '(?<=sglang-kernel==)[0-9A-Za-z\.\-]+' python/pyproject.toml)
     echo "SGL_KERNEL_VERSION_FROM_KERNEL=${SGL_KERNEL_VERSION_FROM_KERNEL} SGL_KERNEL_VERSION_FROM_SRT=${SGL_KERNEL_VERSION_FROM_SRT}"
 
-    if [ "${CUSTOM_BUILD_SGL_KERNEL:-}" = "true" ] && [ -d "sgl-kernel/dist" ]; then
-        ls -alh sgl-kernel/dist
+    if [ "${CUSTOM_BUILD_SGL_KERNEL:-}" = "true" ] && [ -d "python/sglang/kernels/aot/dist" ]; then
+        ls -alh python/sglang/kernels/aot/dist
         if [ "$ARCH" = "aarch64" ] || [ "$ARCH" = "arm64" ]; then
             WHEEL_ARCH="aarch64"
         else
             WHEEL_ARCH="x86_64"
         fi
-        KERNEL_WHL=$(ls sgl-kernel/dist/sglang_kernel-${SGL_KERNEL_VERSION_FROM_KERNEL}+${CU_VERSION}-cp310-abi3-manylinux2014_${WHEEL_ARCH}.whl 2>/dev/null | head -1 || true)
+        KERNEL_WHL=$(ls python/sglang/kernels/aot/dist/sglang_kernel-${SGL_KERNEL_VERSION_FROM_KERNEL}+${CU_VERSION}-cp310-abi3-manylinux2014_${WHEEL_ARCH}.whl 2>/dev/null | head -1 || true)
         if [ -z "$KERNEL_WHL" ]; then
-            echo "ERROR: No matching sgl-kernel wheel found in sgl-kernel/dist/ for version ${SGL_KERNEL_VERSION_FROM_KERNEL} arch ${WHEEL_ARCH} cuda ${CU_VERSION}"
-            ls -alh sgl-kernel/dist/
+            echo "ERROR: No matching sgl-kernel wheel found in python/sglang/kernels/aot/dist/ for version ${SGL_KERNEL_VERSION_FROM_KERNEL} arch ${WHEEL_ARCH} cuda ${CU_VERSION}"
+            ls -alh python/sglang/kernels/aot/dist/
             exit 1
         fi
         echo "Installing sgl-kernel wheel: $KERNEL_WHL"
         $PIP_CMD install "$KERNEL_WHL" --force-reinstall $PIP_INSTALL_SUFFIX
     else
-        if [ "${CUSTOM_BUILD_SGL_KERNEL:-}" = "true" ] && [ ! -d "sgl-kernel/dist" ]; then
-            echo "ERROR: CUSTOM_BUILD_SGL_KERNEL=true but sgl-kernel/dist not found."
+        if [ "${CUSTOM_BUILD_SGL_KERNEL:-}" = "true" ] && [ ! -d "python/sglang/kernels/aot/dist" ]; then
+            echo "ERROR: CUSTOM_BUILD_SGL_KERNEL=true but python/sglang/kernels/aot/dist not found."
             echo "This usually happens when rerunning a stage without the sgl-kernel-build-wheels job."
             echo "Please re-run the full workflow using /tag-and-rerun-ci to rebuild the kernel."
             exit 1

scripts/ci/musa/musa_install_dependency.sh

@@ -84,7 +84,7 @@ else
 
     cd "${REPO_ROOT}" && ${PIP_INSTALL} -v -e "python[dev_musa]" --user
 
-    cd "${REPO_ROOT}/sgl-kernel"
+    cd "${REPO_ROOT}/python/sglang/kernels/aot"
     rm -f pyproject.toml && mv pyproject_musa.toml pyproject.toml && MTGPU_TARGET=mp_31 python3 setup_musa.py install --user
     echo "$HOME/.local/bin" >> "$GITHUB_PATH"
 fi

scripts/ci/musa/rename_wheels_musa.sh

@@ -10,7 +10,7 @@
 # Usage:
 #   rename_wheels_musa.sh <musa_suffix> [wheel_dir]
 # Example:
-#   rename_wheels_musa.sh 43 sgl-kernel/dist
+#   rename_wheels_musa.sh 43 python/sglang/kernels/aot/dist
 set -euxo pipefail
 
 if [[ $# -lt 1 || $# -gt 2 ]]; then

scripts/ci/npu/npu_log_print.sh

@@ -8,7 +8,7 @@ get_version() {
     [ -f "$1" ] && python3 -c 'import re, sys; print(sys.argv[2] + " version: v" + re.search(r"__version__\s*=\s*[\"'"'"'](.*?)[\"'"'"']", open(sys.argv[1]).read()).group(1))' "$1" "$2" 2>/dev/null || echo "$2 version: unknown"
 }
 get_version "./python/sglang/version.py" "sglang"
-get_version "./sgl-kernel/python/sgl_kernel/version.py" "sgl_kernel"
+get_version "./python/sglang/kernels/aot/python/sgl_kernel/version.py" "sgl_kernel"
 SGLANG_URL="https://github.com/sgl-project/sglang.git"
 SGL_KERNEL_URL="https://github.com/sgl-project/sgl-kernel-npu.git"
 SGLANG_BRANCH="main"

scripts/ci/utils/slash_command_handler.py

@@ -289,14 +289,14 @@ def load_permissions(user_login):
 
 def has_sgl_kernel_changes(pr):
     """
-    Check if the PR has changes to the sgl-kernel directory.
+    Check if the PR has changes to the AOT kernel directory.
     This is used to determine if we need a full workflow rerun
     (to rebuild the kernel) vs just rerunning failed jobs.
     """
     try:
         files = pr.get_files()
         for f in files:
-            if f.filename.startswith("sgl-kernel/"):
+            if f.filename.startswith("python/sglang/kernels/aot/"):
                 return True
         return False
     except Exception as e: