[XLA] Fix use-of-unintialized-value msan failure in local_client as well.
PiperOrigin-RevId: 208004791
This commit is contained in:
parent
f2494ede82
commit
b67b3927a1
@ -303,7 +303,7 @@ StatusOr<std::unique_ptr<Literal>> LocalClient::TransferFromOutfeedLocal(
|
||||
const Shape& shape, int device_ordinal) {
|
||||
TF_ASSIGN_OR_RETURN(se::StreamExecutor * executor,
|
||||
backend().stream_executor(device_ordinal));
|
||||
auto literal = MakeUnique<Literal>(shape);
|
||||
auto literal = Literal::CreateFromShape(shape);
|
||||
TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralFromOutfeed(
|
||||
executor, shape, literal.get()));
|
||||
return std::move(literal);
|
||||
|
Loading…
x
Reference in New Issue
Block a user