Update RDTSC benchmark

YaSuenag · Apr 16, 2024 · 4694368 · 4694368
1 parent 19f3d76
commit 4694368
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 14 deletions.
diff --git a/benchmarks/funccall/pom.xml b/benchmarks/funccall/pom.xml
@@ -25,7 +25,7 @@
         <dependency>
             <groupId>com.yasuenag</groupId>
             <artifactId>ffmasm</artifactId>
-            <version>0.2.2</version>
+            <version>0.4.0-SNAPSHOT</version>
         </dependency>
     </dependencies>
 

diff --git a/...rks/funccall/src/main/java/com/yasuenag/ffmasm/benchmark/funccall/FuncCallComparison.java b/...rks/funccall/src/main/java/com/yasuenag/ffmasm/benchmark/funccall/FuncCallComparison.java
@@ -21,6 +21,7 @@ public class FuncCallComparison{
   private CodeSegment seg;
 
   private MethodHandle ffmRDTSC;
+  private MethodHandle ffmRDTSCCritical;
 
   @Setup
   public void setup(){
@@ -39,18 +40,19 @@ public void setup(){
            /* ret            */ .ret()
                                 .getMemorySegment();
 
-      ffmRDTSC = Linker.nativeLinker().downcallHandle(mem, desc, Linker.Option.critical(false));
+      ffmRDTSC = Linker.nativeLinker().downcallHandle(mem, desc);
+      ffmRDTSCCritical = Linker.nativeLinker().downcallHandle(mem, desc, Linker.Option.critical(false));
 
       var register = NativeRegister.create(this.getClass());
-      register.registerNatives(Map.of(this.getClass().getMethod("rdtscFFMDirect"), mem));
+      register.registerNatives(Map.of(this.getClass().getMethod("invokeFFMRDTSCRegisterNatives"), mem));
     }
     catch(Throwable t){
       throw new RuntimeException(t);
     }
   }
 
   @Benchmark
-  public native long rdtsc();
+  public native long invokeJNI();
 
   @Benchmark
   public long invokeFFMRDTSC(){
@@ -63,7 +65,17 @@ public long invokeFFMRDTSC(){
   }
 
   @Benchmark
-  public native long rdtscFFMDirect();
+  public long invokeFFMRDTSCCritical(){
+    try{
+      return (long)ffmRDTSCCritical.invoke();
+    }
+    catch(Throwable t){
+      throw new RuntimeException(t);
+    }
+  }
+
+  @Benchmark
+  public native long invokeFFMRDTSCRegisterNatives();
 
   @TearDown
   public void tearDown(){
@@ -78,13 +90,15 @@ public void tearDown(){
   public static void main(String[] args){
     var inst = new FuncCallComparison();
     inst.setup();
-    long nativeVal = inst.rdtsc();
+    long nativeVal = inst.invokeJNI();
     long ffmVal = inst.invokeFFMRDTSC();
-    long ffmDirectVal = inst.rdtscFFMDirect();
+    long ffmCriticalVal = inst.invokeFFMRDTSCCritical();
+    long ffmRegisterNativesVal = inst.invokeFFMRDTSCRegisterNatives();
 
-    System.out.println("      native: " + nativeVal);
-    System.out.println("         FFM: " + ffmVal);
-    System.out.println("FFM (Direct): " + ffmDirectVal);
+    System.out.println("                  JNI: " + nativeVal);
+    System.out.println("                  FFM: " + ffmVal);
+    System.out.println("       FFM (Critical): " + ffmCriticalVal);
+    System.out.println("FFM (RegisterNatives): " + ffmRegisterNativesVal);
   }
 
 }
diff --git a/benchmarks/funccall/src/main/native/rdtsc.S b/benchmarks/funccall/src/main/native/rdtsc.S
@@ -1,9 +1,9 @@
 .text
 
-# jlong Java_com_yasuenag_ffmasm_benchmark_funccall.FuncCallComparison_rdtsc(JNIEnv *env)
-.global Java_com_yasuenag_ffmasm_benchmark_funccall_FuncCallComparison_rdtsc
-.type   Java_com_yasuenag_ffmasm_benchmark_funccall_FuncCallComparison_rdtsc, @function
-Java_com_yasuenag_ffmasm_benchmark_funccall_FuncCallComparison_rdtsc:
+# jlong Java_com_yasuenag_ffmasm_benchmark_funccall.FuncCallComparison_invokeJNI(JNIEnv *env)
+.global Java_com_yasuenag_ffmasm_benchmark_funccall_FuncCallComparison_invokeJNI
+.type   Java_com_yasuenag_ffmasm_benchmark_funccall_FuncCallComparison_invokeJNI, @function
+Java_com_yasuenag_ffmasm_benchmark_funccall_FuncCallComparison_invokeJNI:
   push %rbp
   mov  %rsp, %rbp
   rdtsc