colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests/end2end/legalize-trigonometric-to-a...

73 lines
2.2 KiB
MLIR
Raw Permalink Normal View History

[MLIR][KernelGen] Legalize `atan2` to approximation Legalize `atan2` analogously to XLA. `atan2` is first reduced to `atan` on the interval [-1, 1] and subsequently approximated. This CL also adds e2e tests for trigonometric approximations. PiperOrigin-RevId: 334794336
2020-10-01 20:33:59 +08:00
// 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
[MLIR] Update tests to eliminate public function declarations. - Also fixed kernel_gen embed_memref_prints to mark the declarations inserted private. - This is in prep for proposed MLIR change to disallow public declarations. PiperOrigin-RevId: 342081252
2020-11-13 02:42:05 +08:00
func private @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface }
[MLIR][KernelGen] Legalize `atan2` to approximation Legalize `atan2` analogously to XLA. `atan2` is first reduced to `atan` on the interval [-1, 1] and subsequently approximated. This CL also adds e2e tests for trigonometric approximations. PiperOrigin-RevId: 334794336
2020-10-01 20:33:59 +08:00
// Helper function to print scalar values.
func @print_f32(%arg : f32) {
Integrate LLVM at llvm/llvm-project@678241795c95 Updates LLVM usage to match [678241795c95](https://github.com/llvm/llvm-project/commit/678241795c95) PiperOrigin-RevId: 363257913
2021-03-17 04:31:59 +08:00
%mem = memref.alloca() : memref<1xf32>
[MLIR][KernelGen] Legalize `atan2` to approximation Legalize `atan2` analogously to XLA. `atan2` is first reduced to `atan` on the interval [-1, 1] and subsequently approximated. This CL also adds e2e tests for trigonometric approximations. PiperOrigin-RevId: 334794336
2020-10-01 20:33:59 +08:00
%c0 = constant 0 : index
Integrate LLVM at llvm/llvm-project@678241795c95 Updates LLVM usage to match [678241795c95](https://github.com/llvm/llvm-project/commit/678241795c95) PiperOrigin-RevId: 363257913
2021-03-17 04:31:59 +08:00
memref.store %arg, %mem[%c0] : memref<1xf32>
%mem_unranked = memref.cast %mem : memref<1xf32> to memref<*xf32>
[MLIR][KernelGen] Legalize `atan2` to approximation Legalize `atan2` analogously to XLA. `atan2` is first reduced to `atan` on the interval [-1, 1] and subsequently approximated. This CL also adds e2e tests for trigonometric approximations. PiperOrigin-RevId: 334794336
2020-10-01 20:33:59 +08:00
call @print_memref_f32(%mem_unranked) : (memref<*xf32>) -> ()
return
}
func @tanh_f32(%arg : f32) {
Integrate LLVM at llvm/llvm-project@2bfe27da171e Updates LLVM usage to match [2bfe27da171e](https://github.com/llvm/llvm-project/commit/2bfe27da171e) PiperOrigin-RevId: 357196336
2021-02-13 00:30:51 +08:00
%res = math.tanh %arg : f32
[MLIR][KernelGen] Legalize `atan2` to approximation Legalize `atan2` analogously to XLA. `atan2` is first reduced to `atan` on the interval [-1, 1] and subsequently approximated. This CL also adds e2e tests for trigonometric approximations. PiperOrigin-RevId: 334794336
2020-10-01 20:33:59 +08:00
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
}