1 //===- RunnerUtils.cpp - Utils for MLIR exec on targets with a C++ runtime ===//
2 //
3 // Part of the LLVM 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 // This file implements basic functions to debug structured MLIR types at
10 // runtime. Entities in this file may not be compatible with targets without a
11 // C++ runtime. These may be progressively migrated to CRunnerUtils.cpp over
12 // time.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #include "mlir/ExecutionEngine/RunnerUtils.h"
17
18 #ifndef _WIN32
19 #include <sys/time.h>
20 #endif // _WIN32
21
_mlir_ciface_print_memref_vector_4x4xf32(StridedMemRefType<Vector2D<4,4,float>,2> * M)22 extern "C" void _mlir_ciface_print_memref_vector_4x4xf32(
23 StridedMemRefType<Vector2D<4, 4, float>, 2> *M) {
24 impl::printMemRef(*M);
25 }
26
_mlir_ciface_print_memref_i8(UnrankedMemRefType<int8_t> * M)27 extern "C" void _mlir_ciface_print_memref_i8(UnrankedMemRefType<int8_t> *M) {
28 impl::printMemRef(*M);
29 }
30
_mlir_ciface_print_memref_i32(UnrankedMemRefType<int32_t> * M)31 extern "C" void _mlir_ciface_print_memref_i32(UnrankedMemRefType<int32_t> *M) {
32 impl::printMemRef(*M);
33 }
34
_mlir_ciface_print_memref_i64(UnrankedMemRefType<int64_t> * M)35 extern "C" void _mlir_ciface_print_memref_i64(UnrankedMemRefType<int64_t> *M) {
36 impl::printMemRef(*M);
37 }
38
_mlir_ciface_print_memref_f32(UnrankedMemRefType<float> * M)39 extern "C" void _mlir_ciface_print_memref_f32(UnrankedMemRefType<float> *M) {
40 impl::printMemRef(*M);
41 }
42
_mlir_ciface_print_memref_f64(UnrankedMemRefType<double> * M)43 extern "C" void _mlir_ciface_print_memref_f64(UnrankedMemRefType<double> *M) {
44 impl::printMemRef(*M);
45 }
46
print_memref_i32(int64_t rank,void * ptr)47 extern "C" void print_memref_i32(int64_t rank, void *ptr) {
48 UnrankedMemRefType<int32_t> descriptor = {rank, ptr};
49 _mlir_ciface_print_memref_i32(&descriptor);
50 }
51
print_memref_i64(int64_t rank,void * ptr)52 extern "C" void print_memref_i64(int64_t rank, void *ptr) {
53 UnrankedMemRefType<int64_t> descriptor = {rank, ptr};
54 _mlir_ciface_print_memref_i64(&descriptor);
55 }
56
print_memref_f32(int64_t rank,void * ptr)57 extern "C" void print_memref_f32(int64_t rank, void *ptr) {
58 UnrankedMemRefType<float> descriptor = {rank, ptr};
59 _mlir_ciface_print_memref_f32(&descriptor);
60 }
61
print_memref_f64(int64_t rank,void * ptr)62 extern "C" void print_memref_f64(int64_t rank, void *ptr) {
63 UnrankedMemRefType<double> descriptor = {rank, ptr};
64 _mlir_ciface_print_memref_f64(&descriptor);
65 }
66
67 extern "C" void
_mlir_ciface_print_memref_0d_f32(StridedMemRefType<float,0> * M)68 _mlir_ciface_print_memref_0d_f32(StridedMemRefType<float, 0> *M) {
69 impl::printMemRef(*M);
70 }
71 extern "C" void
_mlir_ciface_print_memref_1d_f32(StridedMemRefType<float,1> * M)72 _mlir_ciface_print_memref_1d_f32(StridedMemRefType<float, 1> *M) {
73 impl::printMemRef(*M);
74 }
75 extern "C" void
_mlir_ciface_print_memref_2d_f32(StridedMemRefType<float,2> * M)76 _mlir_ciface_print_memref_2d_f32(StridedMemRefType<float, 2> *M) {
77 impl::printMemRef(*M);
78 }
79 extern "C" void
_mlir_ciface_print_memref_3d_f32(StridedMemRefType<float,3> * M)80 _mlir_ciface_print_memref_3d_f32(StridedMemRefType<float, 3> *M) {
81 impl::printMemRef(*M);
82 }
83 extern "C" void
_mlir_ciface_print_memref_4d_f32(StridedMemRefType<float,4> * M)84 _mlir_ciface_print_memref_4d_f32(StridedMemRefType<float, 4> *M) {
85 impl::printMemRef(*M);
86 }
87
88 /// Prints GFLOPS rating.
print_flops(double flops)89 extern "C" void print_flops(double flops) {
90 fprintf(stderr, "%lf GFLOPS\n", flops / 1.0E9);
91 }
92
93 /// Returns the number of seconds since Epoch 1970-01-01 00:00:00 +0000 (UTC).
rtclock()94 extern "C" double rtclock() {
95 #ifndef _WIN32
96 struct timeval tp;
97 int stat = gettimeofday(&tp, NULL);
98 if (stat != 0)
99 fprintf(stderr, "Error returning time from gettimeofday: %d\n", stat);
100 return (tp.tv_sec + tp.tv_usec * 1.0e-6);
101 #else
102 fprintf(stderr, "Timing utility not implemented on Windows\n");
103 return 0.0;
104 #endif // _WIN32
105 }
106