[BugFix] Fix CUDA graph capture for Bounded spec projection #6132
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Build Linux Wheels | |
| on: | |
| pull_request: | |
| push: | |
| branches: | |
| - nightly | |
| - main | |
| - release/* | |
| tags: | |
| # NOTE: Binary build pipelines should only get triggered on release candidate builds | |
| # Release candidate tags look like: v1.11.0-rc1 | |
| - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+ | |
| workflow_dispatch: | |
| workflow_call: | |
| inputs: | |
| test-infra-ref: | |
| description: 'PyTorch test-infra ref (main for nightly, release/X.Y for stable)' | |
| required: false | |
| type: string | |
| default: 'main' | |
| tensordict-source: | |
| description: 'TensorDict source: stable (PyPI), git, or auto (branch-based detection)' | |
| required: false | |
| type: string | |
| default: 'auto' | |
| channel: | |
| description: 'Build channel: nightly, test, release, or empty for auto-detection' | |
| required: false | |
| type: string | |
| default: '' | |
| permissions: | |
| id-token: write | |
| contents: read | |
| env: | |
| # TensorDict source passed to build scripts | |
| # Note: This may not propagate to pytorch/test-infra called workflows; | |
| # scripts also use GITHUB_REF_NAME for branch-based detection as fallback. | |
| TENSORDICT_SOURCE: ${{ inputs.tensordict-source || 'auto' }} | |
| jobs: | |
| generate-matrix: | |
| uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main | |
| with: | |
| package-type: wheel | |
| os: linux | |
| test-infra-repository: pytorch/test-infra | |
| test-infra-ref: ${{ inputs.test-infra-ref || 'main' }} | |
| channel: ${{ inputs.channel || '' }} | |
| use-only-dl-pytorch-org: ${{ inputs.channel == 'release' && 'true' || 'false' }} | |
| build: | |
| needs: generate-matrix | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| include: | |
| - repository: pytorch/rl | |
| smoke-test-script: test/smoke_test.py | |
| package-name: torchrl | |
| pre-script: .github/scripts/pre-build-script.sh | |
| name: pytorch/rl | |
| uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main | |
| with: | |
| repository: ${{ matrix.repository }} | |
| ref: "" | |
| test-infra-repository: pytorch/test-infra | |
| test-infra-ref: ${{ inputs.test-infra-ref || 'main' }} | |
| build-matrix: ${{ needs.generate-matrix.outputs.matrix }} | |
| package-name: ${{ matrix.package-name }} | |
| smoke-test-script: ${{ matrix.smoke-test-script }} | |
| trigger-event: ${{ github.event_name }} | |
| env-var-script: .github/scripts/td_script.sh |