From a3cccd34e2da316e358415f6961b9b9c7e18d233 Mon Sep 17 00:00:00 2001
From: Jevin Jiang <jevinjiang@google.com>
Date: Wed, 28 Aug 2024 15:32:59 -0700
Subject: [PATCH] [Mosaic TPU] Print expected Mosaic version after finding
 unsupported version.

PiperOrigin-RevId: 668632116
---
 jaxlib/mosaic/dialect/tpu/transforms/serde.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/jaxlib/mosaic/dialect/tpu/transforms/serde.cc b/jaxlib/mosaic/dialect/tpu/transforms/serde.cc
index 000bfe3eaea2..3f6050f31dab 100644
--- 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;
       }