[Mosaic TPU] Print expected Mosaic version after finding unsupported …

…version. PiperOrigin-RevId: 667744063
jax-ml · Aug 28, 2024 · 2dfe90f · 2dfe90f
1 parent b010750
commit 2dfe90f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jaxlib/mosaic/dialect/tpu/transforms/serde.cc b/jaxlib/mosaic/dialect/tpu/transforms/serde.cc
@@ -166,8 +166,8 @@ struct MosaicSerdePass : public impl::MosaicSerdePassBase<MosaicSerdePass> {
         return;
       }
       if (version_attr.getInt() > kVersion) {
-        module->emitError("Unsupported Mosaic version: ")
-            << version_attr.getInt();
+        module->emitError("Unsupported Mosaic version:  expected <= ")
+            << kVersion << " but got " << version_attr.getInt();
         signalPassFailure();
         return;
       }