relocate memory stats

SVF-tools · Jan 20, 2024 · e128d68 · e128d68
1 parent a27d116
commit e128d68
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 12 deletions.
diff --git a/svf/lib/Util/PTAStat.cpp b/svf/lib/Util/PTAStat.cpp
@@ -43,6 +43,10 @@ PTAStat::PTAStat(PointerAnalysis* p) : SVFStat(),
                                        _vmsizeUsageBefore(0),
                                        _vmsizeUsageAfter(0)
 {
+    u32_t vmrss = 0;
+    u32_t vmsize = 0;
+    SVFUtil::getMemoryUsageKB(&vmrss, &vmsize);
+    setMemUsageBefore(vmrss, vmsize);
 }
 
 void PTAStat::performStat()
@@ -64,6 +68,13 @@ void PTAStat::performStat()
         }
     }
     PTNumStatMap["LocalVarInRecur"] = localVarInRecursion.count();
+
+    u32_t vmrss = 0;
+    u32_t vmsize = 0;
+    SVFUtil::getMemoryUsageKB(&vmrss, &vmsize);
+    setMemUsageAfter(vmrss, vmsize);
+    PTNumStatMap["MemoryUsageVmrss"] = _vmrssUsageAfter - _vmrssUsageBefore;
+    PTNumStatMap["MemoryUsageVmsize"] = _vmsizeUsageAfter - _vmsizeUsageBefore;
 }
 
 void PTAStat::callgraphStat()

diff --git a/svf/lib/WPA/Andersen.cpp b/svf/lib/WPA/Andersen.cpp
@@ -76,15 +76,9 @@ void AndersenBase::initialize()
     PointerAnalysis::initialize();
     /// Create statistic class
     stat = new AndersenStat(this);
-
-    u32_t vmrss = 0;
-    u32_t vmsize = 0;
-    SVFUtil::getMemoryUsageKB(&vmrss, &vmsize);
-    stat->setMemUsageBefore(vmrss, vmsize);
     /// Build Constraint Graph
     consCG = new ConstraintGraph(pag);
     setGraph(consCG);
-
     if (Options::ConsCGDotGraph())
         consCG->dump("consCG_initial");
 }
@@ -100,12 +94,6 @@ void AndersenBase::finalize()
 
     if (Options::PrintCGGraph())
         consCG->print();
-
-    u32_t vmrss = 0;
-    u32_t vmsize = 0;
-    SVFUtil::getMemoryUsageKB(&vmrss, &vmsize);
-    stat->setMemUsageAfter(vmrss, vmsize);
-
     BVDataPTAImpl::finalize();
 }