Make the XLA CpuTransferManager reflect the fact that the allocator is

synchronous and buffers can be accessed immediately.

PiperOrigin-RevId: 303406287
Change-Id: Ib55caa49b7060e22f9c952cd4d0a7a0ede6dbfb8
This commit is contained in:
A. Unique TensorFlower 2020-03-27 14:22:50 -07:00 committed by TensorFlower Gardener
parent 0f26ae1830
commit 6d732cee5b

View File

@ -44,6 +44,12 @@ class CpuTransferManager : public GenericTransferManager {
const Shape& literal_shape,
MutableBorrowingLiteral literal) override;
bool CanShapedBufferBeAccessedNow(
se::StreamExecutor* executor,
const ShapedBuffer& device_buffer) const override {
return true;
}
private:
Status TransferBufferToInfeed(se::StreamExecutor* executor, int64 size,
const void* source);