Automated Code Change

PiperOrigin-RevId: 688840103
openxla · Oct 23, 2024 · 37635cf · 37635cf
1 parent fd0ce45
commit 37635cf
Show file tree

Hide file tree

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