dd5165a920
Continue the convergence between LLVM dialect and built-in types by replacing the bfloat, half, float and double LLVM dialect types with their built-in counterparts. At the API level, this is a direct replacement. At the syntax level, we change the keywords to `bf16`, `f16`, `f32` and `f64`, respectively, to be compatible with the built-in type syntax. The old keywords can still be parsed but produce a deprecation warning and will be eventually removed. Depends On D94178 Reviewed By: mehdi_amini, silvas, antiagainst Differential Revision: https://reviews.llvm.org/D94179
70 lines
2.3 KiB
MLIR
70 lines
2.3 KiB
MLIR
// RUN: mlir-cpu-runner %s | FileCheck %s
|
|
// RUN: mlir-cpu-runner %s -e foo | FileCheck -check-prefix=NOMAIN %s
|
|
// RUN: mlir-cpu-runner %s --entry-point-result=i32 -e int32_main | FileCheck -check-prefix=INT32MAIN %s
|
|
// RUN: mlir-cpu-runner %s --entry-point-result=i64 -e int64_main | FileCheck -check-prefix=INT64MAIN %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(f32) -> f32
|
|
llvm.func @malloc(i64) -> !llvm.ptr<i8>
|
|
llvm.func @free(!llvm.ptr<i8>)
|
|
|
|
// Check that a simple function with a nested call works.
|
|
llvm.func @main() -> f32 {
|
|
%0 = llvm.mlir.constant(-4.200000e+02 : f32) : f32
|
|
%1 = llvm.call @fabsf(%0) : (f32) -> f32
|
|
llvm.return %1 : f32
|
|
}
|
|
// CHECK: 4.200000e+02
|
|
|
|
// Helper typed functions wrapping calls to "malloc" and "free".
|
|
llvm.func @allocation() -> !llvm.ptr<f32> {
|
|
%0 = llvm.mlir.constant(4 : index) : i64
|
|
%1 = llvm.call @malloc(%0) : (i64) -> !llvm.ptr<i8>
|
|
%2 = llvm.bitcast %1 : !llvm.ptr<i8> to !llvm.ptr<f32>
|
|
llvm.return %2 : !llvm.ptr<f32>
|
|
}
|
|
llvm.func @deallocation(%arg0: !llvm.ptr<f32>) {
|
|
%0 = llvm.bitcast %arg0 : !llvm.ptr<f32> to !llvm.ptr<i8>
|
|
llvm.call @free(%0) : (!llvm.ptr<i8>) -> ()
|
|
llvm.return
|
|
}
|
|
|
|
// Check that allocation and deallocation works, and that a custom entry point
|
|
// works.
|
|
llvm.func @foo() -> f32 {
|
|
%0 = llvm.call @allocation() : () -> !llvm.ptr<f32>
|
|
%1 = llvm.mlir.constant(0 : index) : i64
|
|
%2 = llvm.mlir.constant(1.234000e+03 : f32) : f32
|
|
%3 = llvm.getelementptr %0[%1] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32>
|
|
llvm.store %2, %3 : !llvm.ptr<f32>
|
|
%4 = llvm.getelementptr %0[%1] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32>
|
|
%5 = llvm.load %4 : !llvm.ptr<f32>
|
|
llvm.call @deallocation(%0) : (!llvm.ptr<f32>) -> ()
|
|
llvm.return %5 : f32
|
|
}
|
|
// NOMAIN: 1.234000e+03
|
|
|
|
// Check that i32 return type works
|
|
llvm.func @int32_main() -> i32 {
|
|
%0 = llvm.mlir.constant(42 : i32) : i32
|
|
llvm.return %0 : i32
|
|
}
|
|
// INT32MAIN: 42
|
|
|
|
// Check that i64 return type works
|
|
llvm.func @int64_main() -> i64 {
|
|
%0 = llvm.mlir.constant(42 : i64) : i64
|
|
llvm.return %0 : i64
|
|
}
|
|
// INT64MAIN: 42
|