fix comment, test=develop

4c7b6e2e · liuwei1031 · b20a21e2 · 4c7b6e2e
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/memory/allocation/legacy_allocator.cc paddle/fluid/memory/allocation/legacy_allocator.cc +1 -1

未找到文件。
--- 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() {