| 1372 | : CPUMemoryManager(device, default_memory_pool()) {} |
| 1373 | |
| 1374 | Result<std::unique_ptr<Buffer>> AllocateBuffer(int64_t size) override { |
| 1375 | uint8_t* data; |
| 1376 | RETURN_NOT_OK(pool_->Allocate(size, &data)); |
| 1377 | return std::make_unique<MyBuffer>(data, size, shared_from_this()); |
| 1378 | } |
| 1379 | |
| 1380 | Result<std::shared_ptr<Device::SyncEvent>> MakeDeviceSyncEvent() override { |
| 1381 | return std::make_shared<MyDevice::MySyncEvent>(const_cast<void*>(kMyEventPtr), |