[JAX] Fix error message for matmul operand shape check.

PiperOrigin-RevId: 683778484
jax-ml · Oct 8, 2024 · 9748e2a · 9748e2a
1 parent 2f67710
commit 9748e2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxlib/mosaic/dialect/tpu/transforms/infer_vector_layout.cc b/jaxlib/mosaic/dialect/tpu/transforms/infer_vector_layout.cc
@@ -856,7 +856,7 @@ class VectorLayoutInferer {
       auto shape = vty.getShape().take_back(2);
       if (shape[0] % major_multiple.value_or(tiling[0]) != 0 ||
           shape[1] % minor_multiple.value_or(tiling[1]) != 0) {
-        op->emitOpError("Matmul operand")
+        op->emitOpError("Matmul operand ")
             << operand_name << " must have a shape divisible by ("
             << major_multiple.value_or(tiling[0]) << ", "
             << minor_multiple.value_or(tiling[1]) << "), but got: (" << shape[0]