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

…version. PiperOrigin-RevId: 668632116
jax-ml · Aug 28, 2024 · a3cccd3 · a3cccd3
1 parent b010750
commit a3cccd3
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;
  }