forked from OSchip/llvm-project
33 lines
1.3 KiB
MLIR
33 lines
1.3 KiB
MLIR
|
// RUN: mlir-opt %s -test-math-polynomial-approximation \
|
||
|
|
// RUN: -convert-vector-to-llvm \
|
||
|
|
// RUN: -convert-std-to-llvm \
|
||
|
|
// RUN: | mlir-cpu-runner \
|
||
|
|
// RUN: -e main -entry-point-result=void -O0 \
|
||
|
|
// RUN: -shared-libs=%linalg_test_lib_dir/libmlir_c_runner_utils%shlibext \
|
||
|
|
// RUN: -shared-libs=%linalg_test_lib_dir/libmlir_runner_utils%shlibext \
|
||
|
|
// RUN: | FileCheck %s
|
||
|
|
|
||
|
|
|
||
|
|
func @main() {
|
||
|
|
// ------------------------------------------------------------------------ //
|
||
|
|
// Tanh.
|
||
|
|
// ------------------------------------------------------------------------ //
|
||
|
|
|
||
|
|
// CHECK: 0.848284
|
||
|
|
%0 = constant 1.25 : f32
|
||
|
|
%1 = math.tanh %0 : f32
|
||
|
|
vector.print %1 : f32
|
||
|
|
|
||
|
|
// CHECK: 0.244919, 0.635149, 0.761594, 0.848284
|
||
|
|
%2 = constant dense<[0.25, 0.75, 1.0, 1.25]> : vector<4xf32>
|
||
|
|
%3 = math.tanh %2 : vector<4xf32>
|
||
|
|
vector.print %3 : vector<4xf32>
|
||
|
|
|
||
|
|
// CHECK: 0.099668, 0.197375, 0.291313, 0.379949, 0.462117, 0.53705, 0.604368, 0.664037
|
||
|
|
%4 = constant dense<[0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8]> : vector<8xf32>
|
||
|
|
%5 = math.tanh %4 : vector<8xf32>
|
||
|
|
vector.print %5 : vector<8xf32>
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|