[Feature] Added Lazy implementation of priority updates for replaybuffer prototype #6305
Workflow file for this run
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: | |
| inputs: | |
| build-cpu: | |
| description: 'Build CPU wheels' | |
| required: false | |
| type: boolean | |
| default: true | |
| build-cuda: | |
| description: 'Build CUDA wheels' | |
| required: false | |
| type: boolean | |
| default: false | |
| build-rocm: | |
| description: 'Build ROCm wheels' | |
| required: false | |
| type: boolean | |
| default: false | |
| 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: '' | |
| with-cuda: | |
| description: 'Build with CUDA (enable/disable)' | |
| required: false | |
| type: string | |
| default: 'enable' | |
| with-rocm: | |
| description: 'Build with ROCm (enable/disable)' | |
| required: false | |
| type: string | |
| default: 'enable' | |
| with-cpu: | |
| description: 'Build with CPU (enable/disable)' | |
| required: false | |
| type: string | |
| default: 'enable' | |
| 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' }} | |
| # For workflow_dispatch: convert boolean to enable/disable string | |
| # For workflow_call: use the string input directly | |
| # Default: enable all variants | |
| with-cuda: ${{ github.event_name == 'workflow_dispatch' && (inputs.build-cuda && 'enable' || 'disable') || inputs.with-cuda || 'enable' }} | |
| with-rocm: ${{ github.event_name == 'workflow_dispatch' && (inputs.build-rocm && 'enable' || 'disable') || inputs.with-rocm || 'enable' }} | |
| with-cpu: ${{ github.event_name == 'workflow_dispatch' && (inputs.build-cpu && 'enable' || 'disable') || inputs.with-cpu || 'enable' }} | |
| 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 |