bdusell / semiring-einsum

Generic PyTorch implementation of einsum that supports different semirings
https://bdusell.github.io/semiring-einsum/
MIT License
44 stars 8 forks source link

Doesn't work with PyTorch 2? #43

Closed davidweichiang closed 3 weeks ago

davidweichiang commented 5 months ago

The pyproject.toml (if I understand correctly) requires PyTorch >= 1.1 and < 2.0. Is that necessary?

bdusell commented 5 months ago

PyTorch 2 is backward-compatible with PyTorch 1 from what I understand, so we should fix this.

AutomaticHourglass commented 4 months ago

I also would like to get updated on this. Can someone try this on pytorch 2.0 and run the tests?

bdusell commented 3 weeks ago

Apologies that this took so long -- I've finally fixed this issue, and it will be available in the next release on PyPI!