search

bdusell / semiring-einsum

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

Use minimum block size when memory is not enough #35

Closed ccshan closed 1 year ago

ccshan commented 1 year ago

instead of throwing an exception when choosing an automatic block size

bdusell commented 1 year ago

Looks good! It might be nice to add an option to turn on the old behavior, which could be useful for debugging.