Repository navigation
docs: Update projects #33
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
| # SPDX-FileCopyrightText: 2026 The RISE Project | |
| # SPDX-License-Identifier: MIT | |
| --- | |
| # This workflow is based on: https://github.com/pytorch/tensordict/blob/main/.github/workflows/build-wheels-linux.yml | |
| name: Build tensordict wheels (riscv64) | |
| on: | |
| workflow_dispatch: | |
| inputs: | |
| version: | |
| description: 'Version glob to (re)build; empty builds every version of docs/packages/tensordict.yaml not released yet' | |
| required: false | |
| default: '' | |
| pull_request: | |
| branches: [main] | |
| paths: | |
| - '.github/workflows/build-tensordict.yml' | |
| - 'docs/packages/tensordict.yaml' | |
| - 'patches/tensordict/**' | |
| push: | |
| branches: [main] | |
| paths: | |
| - '.github/workflows/build-tensordict.yml' | |
| - 'docs/packages/tensordict.yaml' | |
| - 'patches/tensordict/**' | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }} | |
| cancel-in-progress: true | |
| permissions: | |
| contents: read # to fetch code (actions/checkout) | |
| jobs: | |
| setup: | |
| uses: $/.github/workflows/_setup.yml | |
| with: | |
| package: tensordict | |
| version: ${{ inputs.version }} | |
| build_wheels: | |
| needs: [setup] | |
| if: needs.setup.outputs.versions != '[]' | |
| name: Build tensordict ${{ matrix.version }} ${{ matrix.python }}-manylinux_riscv64 | |
| runs-on: ubuntu-24.04-riscv | |
| timeout-minutes: 90 | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| version: ${{ fromJSON(needs.setup.outputs.versions) }} | |
| # tensordict's own `torch` dependency has no riscv64 wheel on public PyPI | |
| # (only on pypi.riseproject.dev, for cp312/cp313/cp314/cp314t). | |
| python: | |
| - "cp312" | |
| - "cp313" | |
| - "cp314" | |
| - "cp314t" | |
| env: | |
| TENSORDICT_VERSION: ${{ matrix.version }} | |
| steps: | |
| - name: Checkout tensordict v${{ env.TENSORDICT_VERSION }} | |
| uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1 | |
| with: | |
| repository: pytorch/tensordict | |
| ref: v${{ env.TENSORDICT_VERSION }} | |
| persist-credentials: false | |
| - name: Checkout python-wheels | |
| uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1 | |
| with: | |
| path: python-wheels | |
| persist-credentials: false | |
| - name: Patch tensordict source | |
| run: git apply python-wheels/patches/tensordict/${{ env.TENSORDICT_VERSION }}/00*.patch | |
| - uses: pypa/cibuildwheel@1828c10ab37f080699c7b81cea34097c684a7074 # v4.2.0 | |
| with: | |
| output-dir: wheelhouse/ | |
| only: ${{ matrix.python }}-manylinux_riscv64 | |
| env: | |
| # cmake is a subprocess dependency of setup.py's CMakeBuild, not a | |
| # pyproject.toml build-system requirement, and the manylinux_riscv64 | |
| # image ships no cmake by itself; the PyPI cmake wheel has riscv64 | |
| # builds, so the isolated build env can provision it. | |
| CIBW_BEFORE_BUILD: pip install cmake | |
| # Pins a clean version instead of setuptools_scm's git-derived | |
| # default, which appends `+g<sha>` (setup.py's set_version()). | |
| CIBW_ENVIRONMENT: SETUPTOOLS_SCM_PRETEND_VERSION=${{ env.TENSORDICT_VERSION }} | |
| # torch/numpy/orjson have no riscv64 wheel on public PyPI; resolve | |
| # them from our registry instead, and never fall back to building | |
| # any of the three from source. | |
| CIBW_TEST_ENVIRONMENT: >- | |
| PIP_EXTRA_INDEX_URL=https://pypi.riseproject.dev/simple/ | |
| PIP_ONLY_BINARY=numpy,torch,orjson | |
| CIBW_TEST_COMMAND: python {project}/test/smoke_test.py | |
| - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1 | |
| with: | |
| name: tensordict-${{ env.TENSORDICT_VERSION }}-${{ matrix.python }}-manylinux_riscv64 | |
| path: wheelhouse/*.whl | |
| if-no-files-found: error | |
| publish: | |
| name: Publish tensordict ${{ matrix.version }} | |
| needs: [setup, build_wheels] | |
| if: needs.setup.outputs.versions != '[]' | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| version: ${{ fromJSON(needs.setup.outputs.versions) }} | |
| permissions: | |
| contents: write | |
| pull-requests: write | |
| uses: $/.github/workflows/_publish-wheel.yml | |
| secrets: | |
| app-private-key: ${{ secrets.RISEPROJECT_APP_PRIVATE_KEY }} | |
| with: | |
| artifact-pattern: tensordict-${{ matrix.version }}-*-manylinux_riscv64 |