[Mlir-commits] [mlir] 48d1f09 - [mlir] Add nano precision clock to execution engine

[Denys Shabalin]

Author: Denys Shabalin
Date: 2021-11-09T14:32:36+01:00
New Revision: 48d1f099d492b0d796743d1528f09947e4d2d864

URL: https://github.com/llvm/llvm-project/commit/48d1f099d492b0d796743d1528f09947e4d2d864
DIFF: https://github.com/llvm/llvm-project/commit/48d1f099d492b0d796743d1528f09947e4d2d864.diff

LOG: [mlir] Add nano precision clock to execution engine

Reviewed By: ftynse, nicolasvasilache

Differential Revision: https://reviews.llvm.org/D113476

Added: 
    

Modified: 
    mlir/lib/ExecutionEngine/RunnerUtils.cpp
    mlir/test/python/execution_engine.py

Removed: 
    


################################################################################
diff  --git a/mlir/lib/ExecutionEngine/RunnerUtils.cpp b/mlir/lib/ExecutionEngine/RunnerUtils.cpp
index 83ead7c575a28..3a5f4713c673f 100644
--- a/mlir/lib/ExecutionEngine/RunnerUtils.cpp
+++ b/mlir/lib/ExecutionEngine/RunnerUtils.cpp
@@ -14,6 +14,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "mlir/ExecutionEngine/RunnerUtils.h"
+#include <chrono>
 
 extern "C" void
 _mlir_ciface_print_memref_shape_i8(UnrankedMemRefType<int8_t> *M) {
@@ -75,6 +76,14 @@ extern "C" void _mlir_ciface_print_memref_f64(UnrankedMemRefType<double> *M) {
   impl::printMemRef(*M);
 }
 
+extern "C" int64_t _mlir_ciface_nano_time() {
+  auto now = std::chrono::high_resolution_clock::now();
+  auto duration = now.time_since_epoch();
+  auto nanoseconds =
+      std::chrono::duration_cast<std::chrono::nanoseconds>(duration);
+  return nanoseconds.count();
+}
+
 extern "C" void print_memref_i32(int64_t rank, void *ptr) {
   UnrankedMemRefType<int32_t> descriptor = {rank, ptr};
   _mlir_ciface_print_memref_i32(&descriptor);

diff  --git a/mlir/test/python/execution_engine.py b/mlir/test/python/execution_engine.py
index ffb0de9b69bd2..d83735442c136 100644
--- a/mlir/test/python/execution_engine.py
+++ b/mlir/test/python/execution_engine.py
@@ -358,3 +358,36 @@ def testSharedLibLoad():
 
 
 run(testSharedLibLoad)
+
+
+#  Test that nano time clock is available.
+# CHECK-LABEL: TEST: testNanoTime
+def testNanoTime():
+  with Context():
+    module = Module.parse("""
+      module {
+      func @main() attributes { llvm.emit_c_interface } {
+        %now = call @nano_time() : () -> i64
+        %memref = memref.alloca() : memref<1xi64>
+        %c0 = arith.constant 0 : index
+        memref.store %memref[%c0] : memref<1xi64>
+        call @print_memref_i64(%memref) : (memref<*xi64) -> ()
+        return
+      }
+      func private @nano_time() -> i64 attributes { llvm.emit_c_interface }
+      func private @print_memref_i64(memref<*xi64>) attributes { llvm.emit_c_interface }
+    }""")
+
+    execution_engine = ExecutionEngine(
+        lowerToLLVM(module),
+        opt_level=3,
+        shared_libs=[
+            "../../../../lib/libmlir_runner_utils.so",
+            "../../../../lib/libmlir_c_runner_utils.so"
+        ])
+    execution_engine.invoke("main", arg0_memref_ptr)
+    # CHECK: Unranked Memref
+    # CHECK: [{{.*}}]
+
+
+run(testNanoTime)