Automated Code Change

PiperOrigin-RevId: 713518610
openxla · Jan 10, 2025 · c32f078 · c32f078
1 parent b0a1ac2
commit c32f078
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/xla/pjrt/plugin/xla_gpu/BUILD b/xla/pjrt/plugin/xla_gpu/BUILD
@@ -51,6 +51,7 @@ xla_test(
         ":xla_gpu_pjrt_client",
         "//xla/pjrt/gpu:se_gpu_pjrt_client",
         "//xla/tests:xla_internal_test_main",
+        "@com_google_googletest//:gtest_main",
         "@tsl//tsl/platform:test",
     ],
 )
diff --git a/xla/pjrt/plugin/xla_gpu/xla_gpu_pjrt_client_test.cc b/xla/pjrt/plugin/xla_gpu/xla_gpu_pjrt_client_test.cc
@@ -15,6 +15,8 @@ limitations under the License.
 
 #include "xla/pjrt/plugin/xla_gpu/xla_gpu_pjrt_client.h"
 
+#include <gmock/gmock.h>
+#include <gtest/gtest.h>
 #include "xla/pjrt/gpu/se_gpu_pjrt_client.h"
 #include "tsl/platform/test.h"