Update MemoryUtilTest.java

Over-Run · Feb 24, 2024 · 25e63c5 · 25e63c5
1 parent 0acc5bf
commit 25e63c5
Showing 1 changed file with 28 additions and 28 deletions.
diff --git a/modules/samples/src/jmh/java/overrungl/demo/mem/MemoryUtilTest.java b/modules/samples/src/jmh/java/overrungl/demo/mem/MemoryUtilTest.java
@@ -17,10 +17,12 @@
 package overrungl.demo.mem;
 
 import org.openjdk.jmh.annotations.*;
+import org.openjdk.jmh.infra.Blackhole;
 import org.openjdk.jmh.results.format.ResultFormatType;
 import org.openjdk.jmh.runner.Runner;
 import org.openjdk.jmh.runner.RunnerException;
 import org.openjdk.jmh.runner.options.OptionsBuilder;
+import overrun.marshal.MemoryStack;
 import overrungl.util.MemoryUtil;
 
 import java.lang.foreign.Arena;
@@ -36,57 +38,55 @@
 @BenchmarkMode(Mode.Throughput)
 @State(Scope.Thread)
 @OutputTimeUnit(TimeUnit.SECONDS)
-@Warmup(iterations = 5, time = 1)
-@Measurement(iterations = 5, time = 1)
+@Warmup(iterations = 5, time = 250, timeUnit = TimeUnit.MILLISECONDS, batchSize = 100)
+@Measurement(iterations = 5, time = 250, timeUnit = TimeUnit.MILLISECONDS, batchSize = 100)
 @Threads(Threads.MAX)
 @Fork(1)
 public class MemoryUtilTest {
     @Param({"0", "1", "10", "128", "1024"})
     private long size;
-    private Arena arena;
-
-    @Setup(Level.Iteration)
-    public void setup() {
-        arena = Arena.ofConfined();
-    }
 
     @Benchmark
-    public MemorySegment testArenaInside() {
-        try (Arena arena1 = Arena.ofConfined()) {
-            return arena1.allocate(size);
+    public void measureArena(Blackhole bh) {
+        try (Arena arena = Arena.ofConfined()) {
+            bh.consume(arena.allocate(size));
         }
     }
 
     @Benchmark
-    public MemorySegment testArenaOutside() {
-        return arena.allocate(size);
+    public void measureMalloc(Blackhole bh) {
+        MemorySegment segment = MemorySegment.NULL;
+        try {
+            segment = MemoryUtil.malloc(size);
+            bh.consume(segment);
+        } finally {
+            MemoryUtil.free(segment);
+        }
     }
 
     @Benchmark
-    public MemorySegment memoryUtilMalloc() {
-        MemorySegment seg = null;
+    public void measureCalloc(Blackhole bh) {
+        MemorySegment segment = MemorySegment.NULL;
         try {
-            seg = MemoryUtil.malloc(size);
-            return seg;
+            segment = MemoryUtil.calloc(1L, size);
+            bh.consume(segment);
         } finally {
-            MemoryUtil.free(seg);
+            MemoryUtil.free(segment);
         }
     }
 
     @Benchmark
-    public MemorySegment memoryUtilCalloc() {
-        MemorySegment seg = null;
-        try {
-            seg = MemoryUtil.calloc(1, size);
-            return seg;
-        } finally {
-            MemoryUtil.free(seg);
+    public void measureStackMalloc(Blackhole bh) {
+        try (MemoryStack stack = MemoryStack.stackPush()) {
+            bh.consume(stack.malloc(size, 1L));
         }
     }
 
-    @TearDown(Level.Iteration)
-    public void dispose() {
-        arena.close();
+    @Benchmark
+    public void measureStackCalloc(Blackhole bh) {
+        try (MemoryStack stack = MemoryStack.stackPush()) {
+            bh.consume(stack.allocate(size));
+        }
     }
 
     public static void main(String[] args) throws RunnerException {