File tree Expand file tree Collapse file tree
xla/python/ifrt_proxy/client Expand file tree Collapse file tree Original file line number Diff line number Diff 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
486486cc_library (
Original file line number Diff line number Diff line change 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 (
You can’t perform that action at this time.
0 commit comments