diff --git a/.buildkite/check-wheel-size.py b/.buildkite/check-wheel-size.py index 8178fba552c46..90a5e54736cf3 100644 --- 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 index d79681f03b003..ac60ce0fed14a 100644 --- 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)