How to use Pallas IndexingMode? #22136
Unanswered
Lime-Cakes
asked this question in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
There's currently a lack of documentation on Pallas IndexingMode. There's two mode (Blocked & Unblocked)
jax/jax/_src/pallas/core.py
Line 169 in 945b1c3
However, there doesn't seem to be any documentation on how exactly BlockSpec work in the two modes. Will there be doc for that in the future?
What do they do? And what is their difference?
Edit: I was not aware of #22134 while writing the issue. It contains a lot more doc related to BlockSpec, but doesn't cover Unlocked either.
Beta Was this translation helpful? Give feedback.
All reactions