1 //===- mlir_runner_utils.cpp - Utils for MLIR CPU execution ---------------===//
2 //
3 // Part of the MLIR Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Utilities for interfacing MLIR types with C code as well as printing,
10 // debugging etc.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "include/mlir_runner_utils.h"
15
16 #include <cstdio>
17
18 extern "C" void
print_memref_vector_4x4xf32(StridedMemRefType<Vector2D<4,4,float>,2> * M)19 print_memref_vector_4x4xf32(StridedMemRefType<Vector2D<4, 4, float>, 2> *M) {
20 impl::printMemRef(*M);
21 }
22
23 #define MEMREF_CASE(TYPE, RANK) \
24 case RANK: \
25 impl::printMemRef(*(static_cast<StridedMemRefType<TYPE, RANK> *>(ptr))); \
26 break
27
print_memref_i8(UnrankedMemRefType<int8_t> * M)28 extern "C" void print_memref_i8(UnrankedMemRefType<int8_t> *M) {
29 printUnrankedMemRefMetaData(std::cout, *M);
30 int rank = M->rank;
31 void *ptr = M->descriptor;
32
33 switch (rank) {
34 MEMREF_CASE(int8_t, 0);
35 MEMREF_CASE(int8_t, 1);
36 MEMREF_CASE(int8_t, 2);
37 MEMREF_CASE(int8_t, 3);
38 MEMREF_CASE(int8_t, 4);
39 default:
40 assert(0 && "Unsupported rank to print");
41 }
42 }
43
print_memref_f32(UnrankedMemRefType<float> * M)44 extern "C" void print_memref_f32(UnrankedMemRefType<float> *M) {
45 printUnrankedMemRefMetaData(std::cout, *M);
46 int rank = M->rank;
47 void *ptr = M->descriptor;
48
49 switch (rank) {
50 MEMREF_CASE(float, 0);
51 MEMREF_CASE(float, 1);
52 MEMREF_CASE(float, 2);
53 MEMREF_CASE(float, 3);
54 MEMREF_CASE(float, 4);
55 default:
56 assert(0 && "Unsupported rank to print");
57 }
58 }
59
print_memref_0d_f32(StridedMemRefType<float,0> * M)60 extern "C" void print_memref_0d_f32(StridedMemRefType<float, 0> *M) {
61 impl::printMemRef(*M);
62 }
print_memref_1d_f32(StridedMemRefType<float,1> * M)63 extern "C" void print_memref_1d_f32(StridedMemRefType<float, 1> *M) {
64 impl::printMemRef(*M);
65 }
print_memref_2d_f32(StridedMemRefType<float,2> * M)66 extern "C" void print_memref_2d_f32(StridedMemRefType<float, 2> *M) {
67 impl::printMemRef(*M);
68 }
print_memref_3d_f32(StridedMemRefType<float,3> * M)69 extern "C" void print_memref_3d_f32(StridedMemRefType<float, 3> *M) {
70 impl::printMemRef(*M);
71 }
print_memref_4d_f32(StridedMemRefType<float,4> * M)72 extern "C" void print_memref_4d_f32(StridedMemRefType<float, 4> *M) {
73 impl::printMemRef(*M);
74 }
75
76 // Small runtime support "lib" for vector.print lowering.
77 // By providing elementary printing methods only, this
78 // library can remain fully unaware of low-level implementation
79 // details of our vectors.
print_f32(float f)80 extern "C" void print_f32(float f) { fprintf(stdout, "%g", f); }
print_f64(double d)81 extern "C" void print_f64(double d) { fprintf(stdout, "%lg", d); }
print_open()82 extern "C" void print_open() { fputs("( ", stdout); }
print_close()83 extern "C" void print_close() { fputs(" )", stdout); }
print_comma()84 extern "C" void print_comma() { fputs(", ", stdout); }
print_newline()85 extern "C" void print_newline() { fputc('\n', stdout); }
86