kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[#9632][feat] Support EXTRA_WHEEL_BUILD_ARGS during wheel build (#9633)

Browse Source 
Signed-off-by: Yu Chi Li <yuchil@nvidia.com>
This commit is contained in:
Yu Chi Li 2025-12-02 14:54:58 -08:00 committed by GitHub
parent 0a7a88e74e
commit 7ca38a6c0b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/build_wheel.py
View File

@ -946,8 +946,9 @@ def main(*,
# and validating python changes in the whl.
clear_folder(dist_dir)
extra_wheel_build_args = os.getenv("EXTRA_WHEEL_BUILD_ARGS", "")
build_run(
f'\"{venv_python}\" -m build {project_dir} --skip-dependency-check --no-isolation --wheel --outdir "{dist_dir}"'
f'\"{venv_python}\" -m build {project_dir} --skip-dependency-check {extra_wheel_build_args} --no-isolation --wheel --outdir "{dist_dir}"'
)
if install: