[mlir][runner] Add more printMemref functions.

[bixia]

Summary

Add printMemref for complex data types and index type. Add printMemref for 1d
type beyond f32.

[bixia]

Rebase

[bixia]

Add more printMemref functions.

[aartbik]

mlir/include/mlir/ExecutionEngine/RunnerUtils.h

75

Note that you may want to add a

// Assume index_type is in fact uint64_t.
static_assert(std::is_same<index_type, uint64_t>::value,

"Expected index_type == uint64_t");

in the CPP file for this, so that header and CPP file cannot go out of sync in their view of index_type.

[bixia]

Add static_assert to check index_type assumption.

mlir/include/mlir/ExecutionEngine/RunnerUtils.h

75

Thanks! Added.

mlir/include/mlir/ExecutionEngine/RunnerUtils.h

void printUnrankedMemRefMetaData(StreamType &os, UnrankedMemRefType<T> &v) {
   os << "Unranked MemRef ";
   printMemRefMetaData(os, DynamicMemRefType<T>(v));
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // Templated instantiation follows.
 ////////////////////////////////////////////////////////////////////////////////
 namespace impl {
 template <typename T, int M, int... Dims>
 std::ostream &operator<<(std::ostream &os, const Vector<T, M, Dims...> &v);
 
 template <int... Dims>
 struct StaticSizeMult {
   static constexpr int value = 1;
 };
 
 template <int N, int... Dims>

 _mlir_ciface_printMemref1dF32(StridedMemRefType<float, 1> *m);
 extern "C" MLIR_RUNNERUTILS_EXPORT void
 _mlir_ciface_printMemref2dF32(StridedMemRefType<float, 2> *m);
 extern "C" MLIR_RUNNERUTILS_EXPORT void
 _mlir_ciface_printMemref3dF32(StridedMemRefType<float, 3> *m);
 extern "C" MLIR_RUNNERUTILS_EXPORT void
 _mlir_ciface_printMemref4dF32(StridedMemRefType<float, 4> *m);
 
+extern "C" MLIR_RUNNERUTILS_EXPORT void
+_mlir_ciface_printMemref0dF64(StridedMemRefType<double, 0> *m);
+extern "C" MLIR_RUNNERUTILS_EXPORT void
+_mlir_ciface_printMemref1dF64(StridedMemRefType<double, 1> *m);
+extern "C" MLIR_RUNNERUTILS_EXPORT void
+_mlir_ciface_printMemref2dF64(StridedMemRefType<double, 2> *m);
+extern "C" MLIR_RUNNERUTILS_EXPORT void
+_mlir_ciface_printMemref3dF64(StridedMemRefType<double, 3> *m);
+extern "C" MLIR_RUNNERUTILS_EXPORT void
+_mlir_ciface_printMemref4dF64(StridedMemRefType<double, 4> *m);
+
 extern "C" MLIR_RUNNERUTILS_EXPORT void _mlir_ciface_printMemrefVector4x4xf32(
     StridedMemRefType<Vector2D<4, 4, float>, 2> *m);
 
 extern "C" MLIR_RUNNERUTILS_EXPORT int64_t _mlir_ciface_verifyMemRefI32(
     UnrankedMemRefType<int32_t> *actual, UnrankedMemRefType<int32_t> *expected);
 extern "C" MLIR_RUNNERUTILS_EXPORT int64_t _mlir_ciface_verifyMemRefF32(
     UnrankedMemRefType<float> *actual, UnrankedMemRefType<float> *expected);
 extern "C" MLIR_RUNNERUTILS_EXPORT int64_t _mlir_ciface_verifyMemRefF64(

mlir/lib/ExecutionEngine/RunnerUtils.cpp

 }
 extern "C" void _mlir_ciface_printMemref3dF32(StridedMemRefType<float, 3> *M) {
   impl::printMemRef(*M);
 }
 extern "C" void _mlir_ciface_printMemref4dF32(StridedMemRefType<float, 4> *M) {
   impl::printMemRef(*M);
 }
 
+extern "C" void _mlir_ciface_printMemref0dF64(StridedMemRefType<double, 0> *M) {
+  impl::printMemRef(*M);
+}
+extern "C" void _mlir_ciface_printMemref1dF64(StridedMemRefType<double, 1> *M) {
+  impl::printMemRef(*M);
+}
+extern "C" void _mlir_ciface_printMemref2dF64(StridedMemRefType<double, 2> *M) {
+  impl::printMemRef(*M);
+}
+extern "C" void _mlir_ciface_printMemref3dF64(StridedMemRefType<double, 3> *M) {
+  impl::printMemRef(*M);
+}
+extern "C" void _mlir_ciface_printMemref4dF64(StridedMemRefType<double, 4> *M) {
+  impl::printMemRef(*M);
+}
+
 extern "C" int64_t
 _mlir_ciface_verifyMemRefI32(UnrankedMemRefType<int32_t> *actual,
                              UnrankedMemRefType<int32_t> *expected) {
   return impl::verifyMemRef(*actual, *expected);
 }
 
 extern "C" int64_t
 _mlir_ciface_verifyMemRefF32(UnrankedMemRefType<float> *actual,