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:
parent
0f26ae1830
commit
6d732cee5b
@ -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);
|
||||
|
Loading…
x
Reference in New Issue
Block a user