Automated Code Change

tensorflower-gardener · tensorflower-gardener · commit 191d80dca9e1 · 2024-09-10T23:43:24.000-07:00
PiperOrigin-RevId: 673253634
diff --git a/third_party/xla/xla/backends/interpreter/BUILD b/third_party/xla/xla/backends/interpreter/BUILD
@@ -20,6 +20,7 @@ cc_library(
         "//xla/service:shaped_buffer",
         "//xla/service:transfer_manager",
         "//xla/stream_executor:device_memory",
+        "//xla/stream_executor:stream_executor_h",
     ],
     alwayslink = True,  # Contains per-platform transfer manager registration
 )
diff --git a/third_party/xla/xla/backends/interpreter/executable.cc b/third_party/xla/xla/backends/interpreter/executable.cc
@@ -34,7 +34,6 @@ limitations under the License.
 #include "xla/service/service_executable_run_options.h"
 #include "xla/shape.h"
 #include "xla/shape_util.h"
-#include "xla/stream_executor/stream_executor.h"
 
 namespace xla {
 namespace interpreter {
diff --git a/third_party/xla/xla/backends/interpreter/interpreter_transfer_manager.h b/third_party/xla/xla/backends/interpreter/interpreter_transfer_manager.h
@@ -19,6 +19,7 @@ limitations under the License.
 #include "xla/service/generic_transfer_manager.h"
 #include "xla/service/shaped_buffer.h"
 #include "xla/stream_executor/device_memory.h"
+#include "xla/stream_executor/stream_executor.h"
 
 namespace xla {
 
diff --git a/third_party/xla/xla/backends/interpreter/platform.cc b/third_party/xla/xla/backends/interpreter/platform.cc
@@ -22,6 +22,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/str_format.h"
 #include "xla/backends/interpreter/executor.h"
+#include "xla/stream_executor/device_description.h"
 #include "xla/stream_executor/platform.h"
 #include "xla/stream_executor/platform/initialize.h"
 #include "xla/stream_executor/platform_manager.h"

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,7 @@ cc_library(`
`20`	`20`	`"//xla/service:shaped_buffer",`
`21`	`21`	`"//xla/service:transfer_manager",`
`22`	`22`	`"//xla/stream_executor:device_memory",`
	`23`	`+ "//xla/stream_executor:stream_executor_h",`
`23`	`24`	`],`
`24`	`25`	`alwayslink = True, # Contains per-platform transfer manager registration`
`25`	`26`	`)`