Temporarily remove stablehlo_current_version from PJRT_GetPluginCAttr…

…ibutes.

This can be added back once frameworks release with the fix in:
2f99455

Until then, a plugin that is newer than its framework will error on serialization.

PiperOrigin-RevId: 689524205

xla/pjrt/c/pjrt_c_api_helpers.cc

@@ -634,14 +634,16 @@ static PJRT_NamedValue StableHloVersion(absl::string_view name,
 }
 
 const std::vector<PJRT_NamedValue>& GetXlaPluginCAttributes() {
-  static const std::vector<PJRT_NamedValue>* c_values =
-      new std::vector<PJRT_NamedValue>({
-          XlaVersion("xla_version"),
-          StableHloVersion<0>("stablehlo_current_version",
-                              mlir::vhlo::Version::getCurrentVersion()),
-          StableHloVersion<1>("stablehlo_minimum_version",
-                              mlir::vhlo::Version::getMinimumVersion()),
-      });
+  static const std::vector<PJRT_NamedValue>* c_values = new std::vector<
+      PJRT_NamedValue>({
+      XlaVersion("xla_version"),
+      // TODO: (b/375454646) Uncomment once frameworks have bugfix:
+      // https://github.com/openxla/xla/commit/2f99455cdf99e844ddad17de9f4714997023d243
+      // StableHloVersion<0>("stablehlo_current_version",
+      //                     mlir::vhlo::Version::getCurrentVersion()),
+      StableHloVersion<1>("stablehlo_minimum_version",
+                          mlir::vhlo::Version::getMinimumVersion()),
+  });
   return *c_values;
 }
 

xla/pjrt/c/pjrt_c_api_helpers_test.cc

@@ -213,12 +213,15 @@ TEST(PjRtCApiHelperTest, GetXlaPluginCAttributes) {
     EXPECT_TRUE(did_not_exist_yet);
   }
   EXPECT_TRUE(map.find("xla_version") != map.end());
-  PJRT_NamedValue *current = map["stablehlo_current_version"];
-  mlir::vhlo::Version current_version =
-      mlir::vhlo::Version::getCurrentVersion();
-  EXPECT_TRUE(current->int64_array_value[0] == current_version.getMajor());
-  EXPECT_TRUE(current->int64_array_value[1] == current_version.getMinor());
-  EXPECT_TRUE(current->int64_array_value[2] == current_version.getPatch());
+  // TODO: (b/375454646) Uncomment once frameworks have bugfix:
+  // https://github.com/openxla/xla/commit/2f99455cdf99e844ddad17de9f4714997023d243
+  //
+  // PJRT_NamedValue *current = map["stablehlo_current_version"];
+  // mlir::vhlo::Version current_version =
+  //     mlir::vhlo::Version::getCurrentVersion();
+  // EXPECT_TRUE(current->int64_array_value[0] == current_version.getMajor());
+  // EXPECT_TRUE(current->int64_array_value[1] == current_version.getMinor());
+  // EXPECT_TRUE(current->int64_array_value[2] == current_version.getPatch());
   PJRT_NamedValue *minimum = map["stablehlo_minimum_version"];
   mlir::vhlo::Version minimum_version =
       mlir::vhlo::Version::getMinimumVersion();

xla/pjrt/c/pjrt_c_api_test.cc

@@ -495,7 +495,9 @@ TEST_F(PjrtCApiTest, PluginAttributes) {
     EXPECT_TRUE(did_not_exist_yet);
   }
   EXPECT_TRUE(names.find("xla_version") != names.end());
-  EXPECT_TRUE(names.find("stablehlo_current_version") != names.end());
+  // TODO: (b/375454646) Uncomment once frameworks have bugfix:
+  // https://github.com/openxla/xla/commit/2f99455cdf99e844ddad17de9f4714997023d243
+  // EXPECT_TRUE(names.find("stablehlo_current_version") != names.end());
   EXPECT_TRUE(names.find("stablehlo_minimum_version") != names.end());
 }
 

xla/pjrt/pjrt_c_api_client_test.cc

@@ -170,7 +170,9 @@ TEST(PjRtClientTest, CreateViewAndCopyToDeviceAsyncExternalCpuOnly) {
                                      *literal));
 }
 
-TEST(PjRtClientTest, CompileUsesStableHloVersion) {
+// TODO: (b/375454646) Eanble once frameworks have bugfix:
+// https://github.com/openxla/xla/commit/2f99455cdf99e844ddad17de9f4714997023d243
+TEST(PjRtClientTest, DISABLED_CompileUsesStableHloVersion) {
   SetUpCpuPjRtApi();
   TF_ASSERT_OK_AND_ASSIGN(const PJRT_Api* c_api, pjrt::PjrtApi("cpu"));
   TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<PjRtClient> client,