PyTorch RISC-V support

Hi, it’s Alex from SiFive.

We’ve successfully built PyTorch v2.4.0 for RISC-V on Ubuntu QEMU and have run some AI models. We have some demo snapshots for Stable Diffusion.

Stable Diffusion on RISC-V QEMU:

We’re planning to contribute PyTorch support for RISC-V by submitting patches that include test scripts for CI, Docker images, build scripts, and documentation to guide users in building RISC-V PyTorch.

Any feedback or suggestions for PyTorch RISC-V support?

Hi Alex,

I am interesting in testing this out on keystone riscv. Can you share your build process?

Hi SDK,

The related patches are ongoing. I will update this thread with any updates.