search

iree-org / iree-turbine

IREE's PyTorch Frontend, based on Torch Dynamo.
Apache License 2.0
55 stars 25 forks source link

Restrict to torch < 2.5.0 #228

Closed marbre closed 4 weeks ago

marbre commented 4 weeks ago

The CPU requirements restrict torch to a version < 2.5.0, whereas the ROCm requirements did not so far. The current iree-turbine is incompatible with torch 2.5.0 as torch.export.dynamic_dim() was removed with pytorch/pytorch@b454c51.

Closes #227

Hardcode84 commented 4 weeks ago

Does it work? I've tried this some time ago and they weren't providing 2.4 nightlies anymore.

marbre commented 4 weeks ago

Does it work? I've tried this some time ago and they weren't providing 2.4 nightlies anymore.

I didn't tested for nightlies, but that is what we're doing here already: https://github.com/iree-org/iree-turbine/blob/594d5809e5dee124b154798835436635cb9a4d52/pytorch-cpu-requirements.txt#L3

Hardcode84 commented 4 weeks ago

pytorch-rocm-requirements.txt uses different channel: https://download.pytorch.org/whl/nightly/rocm6.0

marbre commented 4 weeks ago

Seems the main issue is that there are no 2.4.* nighties for ROCm: https://download.pytorch.org/whl/nightly/rocm6.0/torch