提交 4c7b6e2e 编写于 作者: L liuwei1031

fix comment, test=develop

上级 b20a21e2
...@@ -383,7 +383,7 @@ void LegacyMemMonitor::Minus(const int &device, const size_t &size) { ...@@ -383,7 +383,7 @@ void LegacyMemMonitor::Minus(const int &device, const size_t &size) {
uint64_t LegacyMemMonitor::GetMemUsage(const int &device) const { uint64_t LegacyMemMonitor::GetMemUsage(const int &device) const {
return gpu_mem_info_.find(device) == gpu_mem_info_.end() return gpu_mem_info_.find(device) == gpu_mem_info_.end()
? 0 ? 0
: gpu_mem_info_.find(device)->second->GetPeakUsage(); : gpu_mem_info_.at(device)->GetPeakUsage();
} }
void LegacyMemMonitor::PrintMemUsage() { void LegacyMemMonitor::PrintMemUsage() {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册