| 383 | void CheckCPU() const; |
| 384 | |
| 385 | void SetMemoryManager(std::shared_ptr<MemoryManager> mm) { |
| 386 | memory_manager_ = std::move(mm); |
| 387 | is_cpu_ = memory_manager_->is_cpu(); |
| 388 | device_type_ = memory_manager_->device()->device_type(); |
| 389 | } |
| 390 | }; |
| 391 | |
| 392 | /// \defgroup buffer-slicing-functions Functions for slicing buffers |