Hook up memory descriptions extension for TPU.

PiperOrigin-RevId: 712978554
openxla · Jan 7, 2025 · 43e5b9c · 43e5b9c
1 parent 6146cbf
commit 43e5b9c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/xla/pjrt/c/pjrt_c_api_test.cc b/xla/pjrt/c/pjrt_c_api_test.cc
@@ -568,7 +568,6 @@ TEST_F(PjrtCApiTest, DeviceDescriptionAndMemoryDescriptionss) {
 
   for (int i = 0; i < memory_descriptions.size(); i++) {
     EXPECT_NE(memory_descriptions[i].kind().size(), 0);
-    EXPECT_GE(memory_descriptions[i].kind_id(), 0);
   }
 }