Skip to content

Commit 39843a4

Browse files
derekmauroGoogle-ML-Automation
authored andcommitted
Automated Code Change
PiperOrigin-RevId: 903293846
1 parent 60accc8 commit 39843a4

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

xla/python/ifrt_proxy/client/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -480,7 +480,7 @@ cc_library(
480480
"@tsl//tsl/platform:platform_port",
481481
"@tsl//tsl/platform:protobuf",
482482
"@tsl//tsl/profiler/lib:traceme",
483-
],
483+
] + if_google(["//third_party/gloop/strings:cord_bytestream"]),
484484
)
485485

486486
cc_library(

xla/python/ifrt_proxy/client/executable.cc

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,9 @@
3737
#include "absl/synchronization/mutex.h"
3838
#include "absl/time/time.h"
3939
#include "absl/types/span.h"
40+
#if defined(PLATFORM_GOOGLE)
41+
#include "third_party/gloop/strings/cord_bytestream.h"
42+
#endif
4043
#include "llvm/Support/Casting.h"
4144
#include "xla/hlo/ir/hlo_module.h"
4245
#include "xla/layout.h"
@@ -132,7 +135,7 @@ absl::StatusOr<absl::Cord> ExecuteLoadedHostCallback(
132135
std::vector<void*> operand_ptrs;
133136
operand_ptrs.reserve(xla_host_callback.operands.size());
134137

135-
absl::CordReader reader(operand_buffer);
138+
::strings::CordReader reader(operand_buffer);
136139
for (const auto& spec : xla_host_callback.operands) {
137140
const int64_t size = xla::ShapeUtil::ByteSizeOf(spec.shape);
138141
void* p = tsl::port::AlignedMalloc(

0 commit comments

Comments
 (0)