Merge pull request #8068 from The-OpenROAD-Project-staging/mem-stats

Mem stats

Author: maliberty

src/drt/BUILD

@@ -95,7 +95,6 @@ cc_library(
         "src/db/drObj/drVia.cpp",
         "src/db/infra/KDTree.cpp",
         "src/db/infra/frTime.cpp",
-        "src/db/infra/frTime_helper.cpp",
         "src/db/obj/frAccess.cpp",
         "src/db/obj/frInst.cpp",
         "src/db/obj/frInstTerm.cpp",

src/drt/CMakeLists.txt

@@ -63,7 +63,6 @@ add_library(drt_lib
   src/db/drObj/drPin.cpp
   src/db/drObj/drShape.cpp
   src/db/drObj/drVia.cpp
-  src/db/infra/frTime_helper.cpp
   src/db/infra/frTime.cpp
   src/db/infra/KDTree.cpp
   src/db/taObj/taShape.cpp

src/drt/src/db/infra/frTime.cpp

@@ -6,6 +6,8 @@
 #include <boost/io/ios_state.hpp>
 #include <iomanip>
 
+#include "utl/mem_stats.h"
+
 namespace drt {
 
 void frTime::print(Logger* logger)
@@ -30,8 +32,8 @@ void frTime::print(Logger* logger)
                hour,
                min,
                sec,
-               getCurrentRSS() / (1024.0 * 1024.0),
-               getPeakRSS() / (1024.0 * 1024.0));
+               utl::getCurrentRSS() / (1024.0 * 1024.0),
+               utl::getPeakRSS() / (1024.0 * 1024.0));
 }
 
 std::ostream& operator<<(std::ostream& os, const frTime& t)
@@ -50,7 +52,7 @@ std::ostream& operator<<(std::ostream& os, const frTime& t)
   os << ":";
   os << std::setw(2) << std::setfill('0') << sec;
   os << ", memory = " << std::fixed << std::setprecision(2)
-     << getCurrentRSS() * 1.0 / 1024 / 1024 << " (MB)";
+     << utl::getCurrentRSS() * 1.0 / 1024 / 1024 << " (MB)";
   guard.restore();
   return os;
 }

src/drt/src/db/infra/frTime.h

@@ -9,9 +9,6 @@
 
 #include "frBaseTypes.h"
 
-extern size_t getPeakRSS();
-extern size_t getCurrentRSS();
-
 namespace drt {
 class frTime
 {

src/rsz/src/RepairDesign.cc

@@ -31,6 +31,7 @@
 #include "sta/Search.hh"
 #include "sta/SearchPred.hh"
 #include "sta/Units.hh"
+#include "utl/mem_stats.h"
 #include "utl/scope.h"
 
 namespace rsz {
@@ -2652,6 +2653,8 @@ void RepairDesign::printProgress(int iteration,
         inserted_buffer_count_,
         repaired_net_count,
         nets_left);
+
+    debugPrint(logger_, RSZ, "memory", 1, "RSS = {}", utl::getCurrentRSS());
   }
 
   if (end) {

src/rsz/src/RepairHold.cc

@@ -25,6 +25,7 @@
 #include "sta/TimingArc.hh"
 #include "sta/Units.hh"
 #include "utl/Logger.h"
+#include "utl/mem_stats.h"
 
 namespace rsz {
 
@@ -889,6 +890,8 @@ void RepairHold::printProgress(int iteration, bool force, bool end) const
         delayAsString(wns, sta_, 3),
         delayAsString(tns, sta_, 3),
         worst_vertex->name(network_));
+
+    debugPrint(logger_, RSZ, "memory", 1, "RSS = {}", utl::getCurrentRSS());
   }
 
   if (end) {

src/rsz/src/RepairSetup.cc

@@ -38,6 +38,7 @@
 #include "sta/Units.hh"
 #include "sta/VerilogWriter.hh"
 #include "utl/Logger.h"
+#include "utl/mem_stats.h"
 
 namespace rsz {
 
@@ -764,6 +765,8 @@ void RepairSetup::printProgress(const int iteration,
         delayAsString(tns, sta_, 1),
         max(0, num_viols),
         worst_vertex != nullptr ? worst_vertex->name(network_) : "");
+
+    debugPrint(logger_, RSZ, "memory", 1, "RSS = {}", utl::getCurrentRSS());
   }
 
   if (end) {

src/utl/BUILD

@@ -30,6 +30,7 @@ cc_library(
         "src/histogram.cpp",
         "src/prometheus/metrics_server.cpp",
         "src/timer.cpp",
+        "src/mem_stats.cpp",
     ],
     hdrs = [
         "include/utl/CallBack.h",
@@ -66,6 +67,7 @@ cc_library(
         "include/utl/scope.h",
         "include/utl/timer.h",
         "include/utl/validation.h",
+        "include/utl/mem_stats.h",
     ],
     includes = ["include"],
     deps = [

src/utl/CMakeLists.txt

@@ -45,6 +45,7 @@ add_library(utl_lib
   src/Progress.cpp
   src/CommandLineProgress.cpp
   src/timer.cpp
+  src/mem_stats.cpp
   src/decode.cpp
   src/prometheus/metrics_server.cpp
   src/histogram.cpp

src/utl/include/utl/mem_stats.h

@@ -0,0 +1,13 @@
+// SPDX-License-Identifier: BSD-3-Clause
+// Copyright (c) 2019-2025, The OpenROAD Authors
+
+#pragma once
+
+#include <cstddef>
+
+namespace utl {
+
+size_t getPeakRSS();
+size_t getCurrentRSS();
+
+}  // namespace utl