diff --git a/xla/codegen/testlib/BUILD b/xla/codegen/testlib/BUILD
index 833c23e62ffb86..4ba74ccf832b2a 100644
--- a/xla/codegen/testlib/BUILD
+++ b/xla/codegen/testlib/BUILD
@@ -26,6 +26,7 @@ cc_library(
         "//xla:shape_util",
         "//xla:util",
         "//xla:xla_data_proto_cc",
+        "@com_google_absl//absl/log",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/types:span",
         "@tsl//tsl/platform:logging",
diff --git a/xla/codegen/testlib/kernel_runner.cc b/xla/codegen/testlib/kernel_runner.cc
index 7a28469ab9f183..c00b1c5a498dff 100644
--- a/xla/codegen/testlib/kernel_runner.cc
+++ b/xla/codegen/testlib/kernel_runner.cc
@@ -18,6 +18,7 @@ limitations under the License.
 #include <cstddef>
 #include <vector>
 
+#include "absl/log/log.h"
 #include "absl/status/status.h"
 #include "absl/types/span.h"
 #include "xla/literal.h"