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
-
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