[BugFix] Fix DataLoader batching for ragged lazy stacks #2412
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 Aarch64 Linux Wheels | |
| on: | |
| pull_request: | |
| push: | |
| branches: | |
| - nightly | |
| - main | |
| # NOTE: release/* branches are NOT listed here to avoid race conditions | |
| # with the Release workflow which calls this workflow via workflow_call | |
| 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_call: | |
| inputs: | |
| test-infra-ref: | |
| description: 'Branch of pytorch/test-infra to use (e.g., release/2.8 for stable PyTorch)' | |
| required: false | |
| type: string | |
| default: 'main' | |
| nightly: | |
| description: 'Build and publish tensordict-nightly wheels' | |
| required: false | |
| type: boolean | |
| default: false | |
| workflow_dispatch: | |
| permissions: | |
| id-token: write | |
| contents: read | |
| jobs: | |
| generate-matrix: | |
| uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main | |
| with: | |
| package-type: wheel | |
| os: linux-aarch64 | |
| test-infra-repository: pytorch/test-infra | |
| test-infra-ref: ${{ inputs.test-infra-ref || 'main' }} | |
| with-cuda: disable | |
| build: | |
| needs: generate-matrix | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| include: | |
| - repository: pytorch/tensordict | |
| package-name: tensordict | |
| name: pytorch/tensordict | |
| 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 }} | |
| # Not 'schedule' for nightly: that enables test-infra's S3+PyPI upload jobs, | |
| # which need an AWS role tensordict cannot assume; nightly_build.yml uploads | |
| # to PyPI instead. ('' is falsy and would fall through to github.event_name.) | |
| trigger-event: ${{ inputs.nightly && 'skip-upload' || github.event_name }} | |
| env-var-script: .github/scripts/version_script.sh | |
| architecture: aarch64 | |
| setup-miniconda: false | |
| build-command: ${{ inputs.nightly && 'bash .github/scripts/build-aarch64-nightly-wheel.sh' || 'pip wheel --no-deps . && mkdir -p dist && mv tensordict*.whl ./dist' }} | |
| build-platform: python-build-package |