| 1350 | : CPUMemoryManager(device, default_memory_pool()) {} |
| 1351 | |
| 1352 | Result<std::unique_ptr<Buffer>> AllocateBuffer(int64_t size) override { |
| 1353 | uint8_t* data; |
| 1354 | RETURN_NOT_OK(pool_->Allocate(size, &data)); |
| 1355 | return std::make_unique<MyBuffer>(data, size, shared_from_this()); |
| 1356 | } |
| 1357 | |
| 1358 | Result<std::shared_ptr<Device::SyncEvent>> MakeDeviceSyncEvent() override { |
| 1359 | return std::make_shared<MyDevice::MySyncEvent>(const_cast<void*>(kMyEventPtr), |