Pallas+TPU error: Mosaic failed to compile TPU kernel: Couldn't prove that indices divide tiling #22426
Unanswered
Lime-Cakes
asked this question in
Q&A
Pallas+TPU error: Mosaic failed to compile TPU kernel: Couldn't prove that indices divide tiling
#22426
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
While using pl.Unblocked, I keep getting
jaxlib.xla_extension.XlaRuntimeError: INVALID_ARGUMENT: Mosaic failed to compile TPU kernel: Couldn't prove that indices divide tilingerror. I don't understand why/how. What exactly does the error mean?Beta Was this translation helpful? Give feedback.
All reactions