Skip to content

Update build.py to allow customer set the dependency group for wheel … - #8525

Merged
mc-nv merged 1 commit into
mainfrom
mchornyi/PR8329/option
Nov 19, 2025
Merged

Update build.py to allow customer set the dependency group for wheel …#8525
mc-nv merged 1 commit into
mainfrom
mchornyi/PR8329/option

Conversation

@mc-nv

@mc-nv mc-nv commented Nov 19, 2025

Copy link
Copy Markdown
Contributor

This PR provide a change in response on #8329
This change motivated to align with https://peps.python.org/pep-0735/ dependency options.

CPU only build configuration evidences:

# CPU ONLY Scenario
# Dry run on default branch
$ git switch main
$ ./build.py -v --enable-logging --enable-stats --enable-tracing --enable-metrics --filesystem=gcs --filesystem=s3 --filesystem=azure_storage --endpoint=http --endpoint=grpc --endpoint=sagemaker --endpoint=vertex-ai --repo-tag=common:main  --repo-tag=core:main --repo-tag=backend:main --repo-tag=thirdparty:main --backend=ensemble --backend=identity:main --backend=repeat:main --backend=square:main --repoagent=checksum:main --cache=local:main --dryrun ;  grep -e "triton.*whl" build/*
...output omitted...
build/cmake_build:cp /tmp/tritonbuild/tritonserver/install/python/triton*.whl /tmp/tritonbuild/install/python
build/Dockerfile:    "tritonserver-*.whl" | xargs -I {} pip install --upgrade {}[all] && \
build/Dockerfile:    "tritonfrontend-*.whl" | xargs -I {} pip install --upgrade {}[all]
# Dry run on PR branch without options
$ git switch mchornyi/PR8329/option
$ ./build.py -v --enable-logging --enable-stats --enable-tracing --enable-metrics --filesystem=gcs --filesystem=s3 --filesystem=azure_storage --endpoint=http --endpoint=grpc --endpoint=sagemaker --endpoint=vertex-ai --repo-tag=common:main  --repo-tag=core:main --repo-tag=backend:main --repo-tag=thirdparty:main --backend=ensemble --backend=identity:main --backend=repeat:main --backend=square:main --repoagent=checksum:main --cache=local:main --dryrun ;  grep -e "triton.*whl" build/*
...output omitted...
build/cmake_build:cp /tmp/tritonbuild/tritonserver/install/python/triton*.whl /tmp/tritonbuild/install/python
build/Dockerfile:    "tritonserver-*.whl" | xargs -I {} pip install --upgrade {}[all] && \
build/Dockerfile:    "tritonfrontend-*.whl" | xargs -I {} pip install --upgrade {}[all]
# Dry run on PR branch with option: --triton-wheels-dependencies-group=all
$ ./build.py --triton-wheels-dependencies-group=all -v --enable-logging --enable-stats --enable-tracing --enable-metrics --filesystem=gcs --filesystem=s3 --filesystem=azure_storage --endpoint=http --endpoint=grpc --endpoint=sagemaker --endpoint=vertex-ai --repo-tag=common:main  --repo-tag=core:main --repo-tag=backend:main --repo-tag=thirdparty:main --backend=ensemble --backend=identity:main --backend=repeat:main --backend=square:main --repoagent=checksum:main --cache=local:main --dryrun ;  grep -e "triton.*whl" build/*
...output omitted...
build/cmake_build:cp /tmp/tritonbuild/tritonserver/install/python/triton*.whl /tmp/tritonbuild/install/python
build/Dockerfile:    "tritonserver-*.whl" | xargs -I {} pip install --upgrade {}[all] && \
build/Dockerfile:    "tritonfrontend-*.whl" | xargs -I {} pip install --upgrade {}[all]
# Dry run on PR branch with option: --triton-wheels-dependencies-group=cpu
$ ./build.py --triton-wheels-dependencies-group=cpu -v --enable-logging --enable-stats --enable-tracing --enable-metrics --filesystem=gcs --filesystem=s3 --filesystem=azure_storage --endpoint=http --endpoint=grpc --endpoint=sagemaker --endpoint=vertex-ai --repo-tag=common:main  --repo-tag=core:main --repo-tag=backend:main --repo-tag=thirdparty:main --backend=ensemble --backend=identity:main --backend=repeat:main --backend=square:main --repoagent=checksum:main --cache=local:main --dryrun ;  grep -e "triton.*whl" build/*
...output omitted...
build/cmake_build:cp /tmp/tritonbuild/tritonserver/install/python/triton*.whl /tmp/tritonbuild/install/python
build/Dockerfile:    "tritonserver-*.whl" | xargs -I {} pip install --upgrade {}[cpu] && \
build/Dockerfile:    "tritonfrontend-*.whl" | xargs -I {} pip install --upgrade {}[cpu]
# Dry run on PR branch with option: --triton-wheels-dependencies-group=custom
$ ./build.py --triton-wheels-dependencies-group=custom -v --enable-logging --enable-stats --enable-tracing --enable-metrics --filesystem=gcs --filesystem=s3 --filesystem=azure_storage --endpoint=http --endpoint=grpc --endpoint=sagemaker --endpoint=vertex-ai --repo-tag=common:main  --repo-tag=core:main --repo-tag=backend:main --repo-tag=thirdparty:main --backend=ensemble --backend=identity:main --backend=repeat:main --backend=square:main --repoagent=checksum:main --cache=local:main --dryrun ;  grep -e "triton.*whl" build/*
...output omitted...
build/cmake_build:cp /tmp/tritonbuild/tritonserver/install/python/triton*.whl /tmp/tritonbuild/install/python
build/Dockerfile:    "tritonserver-*.whl" | xargs -I {} pip install --upgrade {}[custom] && \
build/Dockerfile:    "tritonfrontend-*.whl" | xargs -I {} pip install --upgrade {}[custom]

@mc-nv
mc-nv requested a review from whoisj November 19, 2025 21:16
@mc-nv mc-nv self-assigned this Nov 19, 2025

@whoisj whoisj left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@mc-nv
mc-nv merged commit db6aa4e into main Nov 19, 2025
3 checks passed
@mc-nv
mc-nv deleted the mchornyi/PR8329/option branch November 19, 2025 22:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Development

Successfully merging this pull request may close these issues.

2 participants