Support non-Darwin, non-Linux, aarch64 platforms

We can read the cntvct_el0 register to observe the CPU clock. It is a single instruction and has barely any overhead.
haskell · May 12, 2024 · acf1cc4 · acf1cc4
1 parent 51e21db
commit acf1cc4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/criterion-measurement/cbits/cycles.c b/criterion-measurement/cbits/cycles.c
@@ -9,6 +9,15 @@ StgWord64 criterion_rdtsc(void)
   return mach_absolute_time();
 }
 
+#elif aarch64_HOST_ARCH
+
+StgWord64 criterion_rdtsc(void)
+{
+  StgWord64 ret;
+  __asm__ __volatile__ ("mrs %0, cntvct_el0" : "=r"(ret));
+  return ret;
+}
+
 #elif x86_64_HOST_ARCH || i386_HOST_ARCH
 
 StgWord64 criterion_rdtsc(void)