mlir-hlo/tests/end2end/legalize-trigonometric-to-a...

73 lines
2.2 KiB
MLIR

// RUN: mlir-hlo-opt %s --mhlo-legalize-trigonometric-to-approximation --convert-std-to-llvm | mlir-cpu-runner -e main -entry-point-result=void --shared-libs=%mlir_runner_utils_dir/libmlir_runner_utils%shlibext | FileCheck %s
func private @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface }
// Helper function to print scalar values.
func @print_f32(%arg : f32) {
%mem = alloca() : memref<1xf32>
%c0 = constant 0 : index
store %arg, %mem[%c0] : memref<1xf32>
%mem_unranked = memref_cast %mem : memref<1xf32> to memref<*xf32>
call @print_memref_f32(%mem_unranked) : (memref<*xf32>) -> ()
return
}
func @tanh_f32(%arg : f32) {
%res = tanh %arg : f32
call @print_f32(%res) : (f32) -> ()
return
}
func @main() {
// Some constants to use as arguments.
%cf_n50_0 = constant -50.0 : f32
%cf_n5_0 = constant -5.0 : f32
%cf_n3_0 = constant -3.0 : f32
%cf_n2_0 = constant -2.0 : f32
%cf_n1_0 = constant -1.0 : f32
%cf_n0_5 = constant -0.5 : f32
%cf_n0_1 = constant -0.1 : f32
%cf_0_0 = constant 0.0 : f32
%cf_0_1 = constant 0.1 : f32
%cf_0_5 = constant 0.5 : f32
%cf_1_0 = constant 1.0 : f32
%cf_2_0 = constant 2.0 : f32
%cf_3_0 = constant 3.0 : f32
%cf_5_0 = constant 5.0 : f32
%cf_50_0 = constant 50.0 : f32
// Tanh.
call @tanh_f32(%cf_n50_0) : (f32) -> ()
// CHECK: -1
call @tanh_f32(%cf_n5_0) : (f32) -> ()
// CHECK: -0.999{{.*}}
call @tanh_f32(%cf_n3_0) : (f32) -> ()
// CHECK: -0.995{{.*}}
call @tanh_f32(%cf_n2_0) : (f32) -> ()
// CHECK: -0.964{{.*}}
call @tanh_f32(%cf_n1_0) : (f32) -> ()
// CHECK: -0.761{{.*}}
call @tanh_f32(%cf_n0_5) : (f32) -> ()
// CHECK: -0.462{{.*}}
call @tanh_f32(%cf_n0_1) : (f32) -> ()
// CHECK: -0.099{{.*}}
call @tanh_f32(%cf_0_0) : (f32) -> ()
// CHECK: 0
call @tanh_f32(%cf_0_1) : (f32) -> ()
// CHECK: 0.099{{.*}}
call @tanh_f32(%cf_0_5) : (f32) -> ()
// CHECK: 0.462{{.*}}
call @tanh_f32(%cf_1_0) : (f32) -> ()
// CHECK: 0.761{{.*}}
call @tanh_f32(%cf_2_0) : (f32) -> ()
// CHECK: 0.964{{.*}}
call @tanh_f32(%cf_3_0) : (f32) -> ()
// CHECK: 0.995{{.*}}
call @tanh_f32(%cf_5_0) : (f32) -> ()
// CHECK: 0.999{{.*}}
call @tanh_f32(%cf_50_0) : (f32) -> ()
// CHECK: 1
return
}