Automated Code Change

PiperOrigin-RevId: 653135460

Author: tensorflower-gardener

third_party/xla/xla/service/gpu/runtime/BUILD

@@ -135,6 +135,7 @@ cc_library(
         ":wait_for_streams_thunk",
         ":while_thunk",
         "//xla:util",
+        "//xla/service:buffer_assignment",
         "//xla/service/gpu/runtime:thunk",
         "@com_google_absl//absl/container:inlined_vector",
         "@com_google_absl//absl/status",

third_party/xla/xla/service/gpu/runtime/command_buffer_cmd_emitter.cc

@@ -23,6 +23,7 @@ limitations under the License.
 #include "absl/container/inlined_vector.h"
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
+#include "xla/service/buffer_assignment.h"
 #include "xla/service/gpu/runtime/command_buffer_cmd.h"
 #include "xla/service/gpu/runtime/conditional_thunk.h"
 #include "xla/service/gpu/runtime/copy_thunk.h"