[mlir] Added support for f64 memref printing in runner utils

Added print_memref_f64 function to runner utils. Differential Revision: https://reviews.llvm.org/D88143
checkedc · Sep 23, 2020 · 5711eaf · 5711eaf
1 parent 34b0848
commit 5711eaf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/mlir/include/mlir/ExecutionEngine/RunnerUtils.h b/mlir/include/mlir/ExecutionEngine/RunnerUtils.h
@@ -220,11 +220,15 @@ extern "C" MLIR_RUNNERUTILS_EXPORT void
 _mlir_ciface_print_memref_i8(UnrankedMemRefType<int8_t> *M);
 extern "C" MLIR_RUNNERUTILS_EXPORT void
 _mlir_ciface_print_memref_f32(UnrankedMemRefType<float> *M);
+extern "C" MLIR_RUNNERUTILS_EXPORT void
+_mlir_ciface_print_memref_f64(UnrankedMemRefType<double> *M);
 
 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_i32(int64_t rank,
                                                          void *ptr);
 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_f32(int64_t rank,
                                                          void *ptr);
+extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_f64(int64_t rank,
+                                                         void *ptr);
 
 extern "C" MLIR_RUNNERUTILS_EXPORT void
 _mlir_ciface_print_memref_0d_f32(StridedMemRefType<float, 0> *M);

diff --git a/mlir/lib/ExecutionEngine/RunnerUtils.cpp b/mlir/lib/ExecutionEngine/RunnerUtils.cpp
@@ -36,6 +36,10 @@ extern "C" void _mlir_ciface_print_memref_f32(UnrankedMemRefType<float> *M) {
   impl::printMemRef(*M);
 }
 
+extern "C" void _mlir_ciface_print_memref_f64(UnrankedMemRefType<double> *M) {
+  impl::printMemRef(*M);
+}
+
 extern "C" void print_memref_i32(int64_t rank, void *ptr) {
   UnrankedMemRefType<int32_t> descriptor = {rank, ptr};
   _mlir_ciface_print_memref_i32(&descriptor);
@@ -46,6 +50,11 @@ extern "C" void print_memref_f32(int64_t rank, void *ptr) {
   _mlir_ciface_print_memref_f32(&descriptor);
 }
 
+extern "C" void print_memref_f64(int64_t rank, void *ptr) {
+  UnrankedMemRefType<double> descriptor = {rank, ptr};
+  _mlir_ciface_print_memref_f64(&descriptor);
+}
+
 extern "C" void
 _mlir_ciface_print_memref_0d_f32(StridedMemRefType<float, 0> *M) {
   impl::printMemRef(*M);