maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[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
This commit is contained in:
Jakub Lichman 2020-09-23 11:52:10 +00:00
parent 34b08487f0
commit 5711eaf608
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/include/mlir/ExecutionEngine/RunnerUtils.h
View File

@ -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);

9
mlir/lib/ExecutionEngine/RunnerUtils.cpp
View File

@ -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);