simple.mlir
1.91 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// RUN: mlir-cpu-runner %s | FileCheck %s
// RUN: mlir-cpu-runner %s -e foo | FileCheck -check-prefix=NOMAIN %s
// RUN: mlir-cpu-runner %s -O3 | FileCheck %s
// RUN: cp %s %t
// RUN: mlir-cpu-runner %t -dump-object-file | FileCheck %t
// RUN: ls %t.o
// RUN: rm %t.o
// RUN: mlir-cpu-runner %s -dump-object-file -object-filename=%T/test.o | FileCheck %s
// RUN: ls %T/test.o
// RUN: rm %T/test.o
// Declarations of C library functions.
llvm.func @fabsf(!llvm.float) -> !llvm.float
llvm.func @malloc(!llvm.i64) -> !llvm<"i8*">
llvm.func @free(!llvm<"i8*">)
// Check that a simple function with a nested call works.
llvm.func @main() -> !llvm.float {
%0 = llvm.mlir.constant(-4.200000e+02 : f32) : !llvm.float
%1 = llvm.call @fabsf(%0) : (!llvm.float) -> !llvm.float
llvm.return %1 : !llvm.float
}
// CHECK: 4.200000e+02
// Helper typed functions wrapping calls to "malloc" and "free".
llvm.func @allocation() -> !llvm<"float*"> {
%0 = llvm.mlir.constant(4 : index) : !llvm.i64
%1 = llvm.call @malloc(%0) : (!llvm.i64) -> !llvm<"i8*">
%2 = llvm.bitcast %1 : !llvm<"i8*"> to !llvm<"float*">
llvm.return %2 : !llvm<"float*">
}
llvm.func @deallocation(%arg0: !llvm<"float*">) {
%0 = llvm.bitcast %arg0 : !llvm<"float*"> to !llvm<"i8*">
llvm.call @free(%0) : (!llvm<"i8*">) -> ()
llvm.return
}
// Check that allocation and deallocation works, and that a custom entry point
// works.
llvm.func @foo() -> !llvm.float {
%0 = llvm.call @allocation() : () -> !llvm<"float*">
%1 = llvm.mlir.constant(0 : index) : !llvm.i64
%2 = llvm.mlir.constant(1.234000e+03 : f32) : !llvm.float
%3 = llvm.getelementptr %0[%1] : (!llvm<"float*">, !llvm.i64) -> !llvm<"float*">
llvm.store %2, %3 : !llvm<"float*">
%4 = llvm.getelementptr %0[%1] : (!llvm<"float*">, !llvm.i64) -> !llvm<"float*">
%5 = llvm.load %4 : !llvm<"float*">
llvm.call @deallocation(%0) : (!llvm<"float*">) -> ()
llvm.return %5 : !llvm.float
}
// NOMAIN: 1.234000e+03