Open
Milestone

Build pytorch test dependencies

We already have all the dependencies needed to build and run pytorch on riscv64, but we are missing some of the dependencies needed to run all the pytorch test cases. Here are a list of the missing dependencies.

onnx==1.17.0
PyNaCl==1.5.0
aiohttp==3.11.16
grpcio==1.71.0
ml_dtypes==0.5.1
pywavelets==1.7.0
pyzstd==0.16.2
scikit_image==0.22.0
scipy==1.14.1
tensorboard_data_server==0.7.2
wrapt==1.17.2
z3_solver==4.12.6.0
pyzstd-0.16.2

These dependencies either come from the .ci/docker/requirements-ci.txt file or are dependencies of the packages listed in this file. Some of the required packages are also needed by vllm and are covered in the vllm milestone. As we cannot add issues to multiple milestones and we cannot make one milestone depend on another, we'll just have to state here that the vllm milestone must be completed before we can consider this milestone to be complete.

grpcio and PyNACL are part of the Build the top 30 Python packages Milestone so is also not explicitly associated with this milestone.

PyTorch 2.7 cpu dependency tree

torch==2.7.0+cpu
├── filelock [required: Any, installed: 3.13.1]
├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
├── setuptools [required: Any, installed: 70.2.0]
├── sympy [required: >=1.13.3, installed: 1.13.3]
│   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
├── networkx [required: Any, installed: 3.3]
├── Jinja2 [required: Any, installed: 3.1.4]
│   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
└── fsspec [required: Any, installed: 2024.6.1]

To capture this dependency tree, run the following:

python3 -m venv .venv
source .venv/bin/activate
pip install --upgrade pip
pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu
pip install pipdeptree
pipdeptree
  • Work items 5
  • Merge requests 0
  • Participants 2
  • Labels 1
Loading
Loading
Loading
Loading
100% complete
100%
Start date
No start date
None
Due date
No due date
5
Work items 5 New issue
Open: 0 Closed: 5
None
Total weight
None
1
Merge requests 1
Open: 0 Closed: 0 Merged: 1
0
Releases
None
Reference: riseproject/python/wheel_builder%"Build pytorch test dependencies"