fix comment, test=develop

6 years ago · 4c7b6e2e67
parent b20a21e299
commit 4c7b6e2e67
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/memory/allocation/legacy_allocator.cc
+++ b/paddle/fluid/memory/allocation/legacy_allocator.cc
@ -383,7 +383,7 @@ void LegacyMemMonitor::Minus(const int &device, const size_t &size) {
 uint64_t LegacyMemMonitor::GetMemUsage(const int &device) const {
  return gpu_mem_info_.find(device) == gpu_mem_info_.end()
             ? 0
-             : gpu_mem_info_.find(device)->second->GetPeakUsage();
+             : gpu_mem_info_.at(device)->GetPeakUsage();
 }

 void LegacyMemMonitor::PrintMemUsage() {