[CI] Reduce wheel size by not shipping debug symbols (#4602)

vllm-project · May 5, 2024 · c7f2cf2 · c7f2cf2
1 parent 8d8357c
commit c7f2cf2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/.buildkite/check-wheel-size.py b/.buildkite/check-wheel-size.py
@@ -25,6 +25,9 @@ def check_wheel_size(directory):
  f"compare to the allowed size ({MAX_SIZE_MB} MB).")
  print_top_10_largest_files(wheel_path)
  return 1
+ else:
+ print(f"Wheel {wheel_path} is within the allowed size "
+ f"({wheel_size_mb} MB).")
  return 0
 
 

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -79,6 +79,8 @@ jobs:
 
  - name: Build wheel
  shell: bash
+ env:
+ CMAKE_BUILD_TYPE: Release # do not compile with debug symbol to reduce wheel size
  run: |
  bash -x .github/workflows/scripts/build.sh ${{ matrix.python-version }} ${{ matrix.cuda-version }}
  wheel_name=$(ls dist/*whl | xargs -n 1 basename)