speed up sgl-kernel build (#18586)

2026-02-12 23:43:22 +08:00
parent 2f283d8152
commit 9e9e949261
4 changed files with 94 additions and 8 deletions
--- a/.github/workflows/release-whl-kernel.yml
+++ b/.github/workflows/release-whl-kernel.yml
@@ -46,7 +46,8 @@ jobs:
          chmod +x ./build.sh
          ./build.sh "${{ matrix.python-version }}" "${{ matrix.cuda-version }}" ${{ matrix.arch == 'aarch64' && 'aarch64' || '' }}
        env:
-          USE_CCACHE: 0
+          BUILD_JOBS: 64
+          NVCC_THREADS: 8

      - name: Upload to PyPI
        working-directory: sgl-kernel
@@ -139,7 +140,8 @@ jobs:
          chmod +x ./build.sh
          ./build.sh "${{ matrix.python-version }}" "${{ matrix.cuda-version }}" ${{ matrix.arch == 'aarch64' && 'aarch64' || '' }}
        env:
-          USE_CCACHE: 0
+          BUILD_JOBS: 64
+          NVCC_THREADS: 8

      - name: Upload artifacts
        uses: actions/upload-artifact@v4