Temp for debugging excessive memory usage

NadavKeren · Dec 25, 2024 · 5aebfb4 · 5aebfb4
1 parent f0184bc
commit 5aebfb4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/simulator/build.gradle b/simulator/build.gradle
@@ -82,7 +82,7 @@ tasks.named('run').configure {
     it.getKey().startsWith('akka') || it.getKey().startsWith('caffeine')
   }
 
-  jvmArgs '-XX:+UseParallelGC', '-Xmx4g'
+  jvmArgs '-XX:+UseParallelGC', '-Xmx14g'
   if (project.hasProperty('jvmArgs')) {
     jvmArgs project.jvmArgs.split(',')
   }

diff --git a/simulator/src/main/java/com/github/benmanes/caffeine/cache/simulator/policy/PolicyActor.java b/simulator/src/main/java/com/github/benmanes/caffeine/cache/simulator/policy/PolicyActor.java
@@ -39,6 +39,9 @@ public final class PolicyActor extends AbstractBehavior<PolicyActor.Command> {
   private final Policy policy;
   private int eventNumber = 0;
 
+  private static final long MEMORY_THERSHOLD_GIGA = 4;
+  private static final long GIGABYTE = 1073741824L;
+
   public PolicyActor(ActorContext<Command> context,
       ActorRef<Simulator.Command> simulator, Policy policy) {
     super(context);
@@ -59,11 +62,19 @@ public Receive<Command> createReceive() {
   }
 
   private Behavior<Command> process(List<AccessEvent> events) {
+    Runtime runtime = Runtime.getRuntime();
     policy.stats().stopwatch().start();
     for (AccessEvent event : events) {
       event.setEventNum(eventNumber++);
       try {
         policy.record(event);
+        if (eventNumber % 1000000 == 0) {
+          double currentRAMUsage = (double) runtime.totalMemory() / GIGABYTE;
+          System.out.println(eventNumber / 1000000 + "m: Memory Used: " + currentRAMUsage + " GB");
+          if (currentRAMUsage > MEMORY_THERSHOLD_GIGA) {
+            System.out.println("Exceeded the threshold");
+          }
+        }
       } catch (RuntimeException e) {
         System.err.println("Error on event: " + event.eventNum());
         e.printStackTrace();

diff --git a/simulator/src/main/java/com/github/benmanes/caffeine/cache/simulator/policy/Registry.java b/simulator/src/main/java/com/github/benmanes/caffeine/cache/simulator/policy/Registry.java
@@ -219,7 +219,6 @@ private void registerSketch() {
     registerMany(GhostHillClimberTinyLfuPolicy.class, GhostHillClimberTinyLfuPolicy::policies);
     registerMany(AdaptiveCAPolicy.class, AdaptiveCAPolicy::policies);
     register(AdaptiveCAWithBurstBlockPolicy.class, AdaptiveCAWithBurstBlockPolicy::policy);
-    register(AdaptivePipelineCache.class, AdaptivePipelineCache::new);
 
     register(TinyCachePolicy.class, TinyCachePolicy::new);
     register(WindowTinyCachePolicy.class, WindowTinyCachePolicy::new);