[mlir] Add print functions for i64, index.

[tpopp]

mlir/lib/ExecutionEngine/RunnerUtils.cpp

48

Just confirming that this shouldn't be something like size_t.

[herhut]

mlir/include/mlir/ExecutionEngine/RunnerUtils.h

226

I would not add index, as we do not know at this level (C, which is called from LLVM) what type index will have. Could we instead do something where we call the print_memref_i32 and have a index_cast operation in front? Not sure index_cast works on memrefs but that would be a nice approach.

Until then, maybe at least call this print_memref_index_as_i64?

mlir/lib/ExecutionEngine/RunnerUtils.cpp

48

Directly forward to`print_memref_i64`?

[pifon2a]

Address the comments.

mlir/include/mlir/ExecutionEngine/RunnerUtils.h

 ////////////////////////////////////////////////////////////////////////////////
 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_index(int64_t rank,
+                                                           void *ptr);
 extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_i32(int64_t rank,
                                                          void *ptr);
+extern "C" MLIR_RUNNERUTILS_EXPORT void print_memref_i64(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);
 extern "C" MLIR_RUNNERUTILS_EXPORT void

mlir/lib/ExecutionEngine/RunnerUtils.cpp

 extern "C" void _mlir_ciface_print_memref_i8(UnrankedMemRefType<int8_t> *M) {
   impl::printMemRef(*M);
 }
 
 extern "C" void _mlir_ciface_print_memref_i32(UnrankedMemRefType<int32_t> *M) {
   impl::printMemRef(*M);
 }
 
+extern "C" void _mlir_ciface_print_memref_i64(UnrankedMemRefType<int64_t> *M) {
+  impl::printMemRef(*M);
+}
+
 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_index(int64_t rank, void *ptr) {
+  UnrankedMemRefType<int64_t> descriptor = {rank, ptr};
+  _mlir_ciface_print_memref_i64(&descriptor);
+}
+
 extern "C" void print_memref_i32(int64_t rank, void *ptr) {
   UnrankedMemRefType<int32_t> descriptor = {rank, ptr};
   _mlir_ciface_print_memref_i32(&descriptor);
 }
 
+extern "C" void print_memref_i64(int64_t rank, void *ptr) {
+  UnrankedMemRefType<int64_t> descriptor = {rank, ptr};
+  _mlir_ciface_print_memref_i64(&descriptor);
+}
+
 extern "C" void print_memref_f32(int64_t rank, void *ptr) {
   UnrankedMemRefType<float> descriptor = {rank, 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);