Skip to content

tensordict: add build-tensordict.yml for riscv64 wheels - #1868

Merged
luhenry merged 1 commit into
mainfrom
tensordict
Sep 14, 2026
Merged

tensordict: add build-tensordict.yml for riscv64 wheels#1868
luhenry merged 1 commit into
mainfrom
tensordict

Conversation

@luhenry

@luhenry luhenry commented Sep 12, 2026

Copy link
Copy Markdown
Member

Compiles a small pybind11 extension (tensordict._C, key-unraveling helpers) that does not link libtorch; torch is a pure runtime dependency, already published on pypi.riseproject.dev. Upstream publishes no riscv64 wheel.

Mirrors upstream's build-wheels-linux.yml, narrowed to a plain cibuildwheel build since upstream's own pytorch/test-infra reusable workflow doesn't apply here.

Differs from upstream

  • Sets SETUPTOOLS_SCM_PRETEND_VERSION - pins a clean wheel version instead of setuptools_scm's git-derived +g<sha> default.
  • pip install cmake in before-build - setup.py shells out to cmake directly, not a pyproject.toml build dependency.

Matrix: cp312/cp313/cp314/cp314t only - torch has no riscv64 wheel on pypi.riseproject.dev for cp310/cp311.

Testing

  • same as upstream (test/smoke_test.py)
  • torch/numpy/orjson resolved from pypi.riseproject.dev; public PyPI ships no riscv64 wheel for any of the three.

License: OK

Patches

  • 0001-csrc-request-only-Development.Module-from-FindPython.patch - To upstream (not submitted; this repo does not open issues/PRs on third-party repositories). CMake's Development component pulls in Development.Embed, which fails on manylinux's static-libpython builds; an extension module only needs Development.Module. Reproduces on manylinux x86_64/aarch64 too, not just riscv64.

Built on cp312; smoke test passed.

luhenry added a commit that referenced this pull request Sep 12, 2026
@luhenry
luhenry merged commit f91cea7 into main Sep 14, 2026
9 checks passed
@luhenry
luhenry deleted the tensordict branch September 14, 2026 09:28
@luhenry luhenry linked an issue Sep 14, 2026 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

tensordict riscv64 support

1 participant