jax

88af892f - [Mosaic TPU] Explicitly specify the padding value in vector.transfer_read

[Mosaic TPU] Explicitly specify the padding value in vector.transfer_read The default builder has recently changed in upstream MLIR, where instead of using a constant 0, the padding value changes to `ub.poison`. UB is not a dialect we use in our compiler, which unsurprisingly leads to failures down the line. See https://github.com/llvm/llvm-project/commit/878d3594ed74cd1153bc5cba2cb7a449702cdf34 for reference. The way we use `vector.transfer_read` never involves any padding, so the value doesn't matter. We now explicitly specify 0. PiperOrigin-RevId: 781510446