Automated Code Change

PiperOrigin-RevId: 728144120
openxla · Feb 18, 2025 · 6b7233b · 6b7233b
1 parent 23e8031
commit 6b7233b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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 {