[Mosaic] Improves error messages for infer vector layout.

PiperOrigin-RevId: 604697612
jax-ml · Feb 6, 2024 · 87d1670 · 87d1670
1 parent 9a098e9
commit 87d1670
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/jaxlib/mosaic/dialect/tpu/transforms/infer_vector_layout.cc b/jaxlib/mosaic/dialect/tpu/transforms/infer_vector_layout.cc
@@ -435,6 +435,7 @@ class VectorLayoutInferer {
       setLayout(op, src_layout, dst_layout);
       return success();
     }
+    op.emitOpError("unsupported truncation layout");
     return failure();
   }
 
@@ -494,6 +495,7 @@ class VectorLayoutInferer {
       setLayout(op, src_layout, dst_layout);
       return success();
     }
+    op.emitOpError("unsupported truncation layout");
     return failure();
   }