MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TotalMem

Method TotalMem

tensorflow/core/common_runtime/gpu_memory_planner.cc:277–290  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

275}
276
277size_t GPULifetimePolicy::TotalMem() const {
278 size_t total_mem = 0;
279 for (auto bin : bins_) {
280 total_mem += bin->TotalMem();
281 }
282 {
283 std::lock_guard<spin_lock> l(large_bin_lock_);
284 for (auto large_bin : large_bins_) {
285 auto bin_info = large_bin.second;
286 total_mem += bin_info->TotalMem();
287 }
288 }
289 return total_mem;
290}
291
292void GPULifetimePolicy::Dump() const {
293 // LOG(INFO_DEV) << "GPULifetimePolicy, start:" << start_

Callers 1

BestLifetimePolicyMethod · 0.45

Calls 2

RoundedBytesFunction · 0.70
sizeMethod · 0.45

Tested by

no test coverage detected