| 207 | } |
| 208 | |
| 209 | void StreamExecutor::Deallocate(DeviceMemoryBase *mem) { |
| 210 | VLOG(1) << "Called StreamExecutor::Deallocate(mem=" << mem->opaque() |
| 211 | << ") mem->size()=" << mem->size() << StackTraceIfVLOG10(); |
| 212 | |
| 213 | if (mem->opaque() != nullptr) { |
| 214 | EraseAllocRecord(mem->opaque()); |
| 215 | } |
| 216 | implementation_->Deallocate(mem); |
| 217 | mem->Reset(nullptr, 0); |
| 218 | } |
| 219 | |
| 220 | void StreamExecutor::GetMemAllocs(std::map<void *, AllocRecord> *records_out) { |
| 221 | absl::ReaderMutexLock lock(&mu_); |
nothing calls this directly
no test coverage detected