From 37635cf0e622f09bcef85cc09b9c1eb4ecea9987 Mon Sep 17 00:00:00 2001 From: xla authors Date: Tue, 22 Oct 2024 23:54:19 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 688840103 --- xla/stream_executor/BUILD | 1 + xla/stream_executor/dnn.cc | 2 ++ xla/stream_executor/dnn.h | 1 + 3 files changed, 4 insertions(+) diff --git a/xla/stream_executor/BUILD b/xla/stream_executor/BUILD index 8a2969a44ba95..458b4766afe0c 100644 --- a/xla/stream_executor/BUILD +++ b/xla/stream_executor/BUILD @@ -456,6 +456,7 @@ cc_library( "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/types:span", "@eigen_archive//:eigen3", # buildcleaner: keep + "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:ml_dtypes", "@tsl//tsl/platform:status", diff --git a/xla/stream_executor/dnn.cc b/xla/stream_executor/dnn.cc index d59080deeb694..6b7a87d80b3ae 100644 --- a/xla/stream_executor/dnn.cc +++ b/xla/stream_executor/dnn.cc @@ -41,6 +41,8 @@ limitations under the License. #include "xla/stream_executor/data_type.h" #include "xla/stream_executor/device_memory.h" #include "xla/stream_executor/numeric_options.h" +#include "xla/stream_executor/scratch_allocator.h" +#include "xla/stream_executor/stream.h" #include "xla/tsl/lib/strings/proto_serialization.h" #include "xla/tsl/protobuf/dnn.pb.h" #include "xla/util.h" diff --git a/xla/stream_executor/dnn.h b/xla/stream_executor/dnn.h index 4103b6d6f7185..f99f6a0380c39 100644 --- a/xla/stream_executor/dnn.h +++ b/xla/stream_executor/dnn.h @@ -47,6 +47,7 @@ limitations under the License. #include "xla/stream_executor/scratch_allocator.h" #include "xla/stream_executor/stream.h" #include "xla/tsl/protobuf/dnn.pb.h" +#include "tsl/platform/errors.h" #include "tsl/platform/logging.h" namespace Eigen {