From 6b7233b8189162c003c92f69986472f1872d362e Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Tue, 18 Feb 2025 04:07:03 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 728144120
---
 xla/service/gpu/tests/BUILD                       | 1 +
 xla/service/gpu/tests/async_kernel_launch_test.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/xla/service/gpu/tests/BUILD b/xla/service/gpu/tests/BUILD
index 1b24e973795d1..9705f81603243 100644
--- a/xla/service/gpu/tests/BUILD
+++ b/xla/service/gpu/tests/BUILD
@@ -124,6 +124,7 @@ xla_cc_test(
         "//xla:debug_options_flags",
         "//xla:literal",
         "//xla:literal_util",
+        "//xla:xla_proto_cc",
         "//xla/service:gpu_plugin",
         "//xla/service:hlo_module_config",
         "//xla/tests:hlo_test_base",
diff --git a/xla/service/gpu/tests/async_kernel_launch_test.cc b/xla/service/gpu/tests/async_kernel_launch_test.cc
index e5ff2b1570167..7aac65a38121c 100644
--- a/xla/service/gpu/tests/async_kernel_launch_test.cc
+++ b/xla/service/gpu/tests/async_kernel_launch_test.cc
@@ -22,6 +22,7 @@ limitations under the License.
 #include "xla/service/hlo_module_config.h"
 #include "xla/tests/hlo_test_base.h"
 #include "xla/tests/literal_test_util.h"
+#include "xla/xla.pb.h"
 
 namespace xla::gpu {
 namespace {