2f5715dc78
The remanants of Standard was renamed to Func, but the test directory remained named as Standard. In adidition to fixing the name, this commit also moves the tests for operations not in the Func dialect to the proper parent dialect test directory.
24 lines
1.1 KiB
MLIR
24 lines
1.1 KiB
MLIR
// RUN: mlir-opt %s -test-expand-tanh | FileCheck %s
|
|
|
|
// CHECK-LABEL: func @tanh
|
|
func @tanh(%arg: f32) -> f32 {
|
|
%res = math.tanh %arg : f32
|
|
return %res : f32
|
|
}
|
|
// CHECK-DAG: %[[ZERO:.+]] = arith.constant 0.000000e+00 : f32
|
|
// CHECK-DAG: %[[ONE:.+]] = arith.constant 1.000000e+00 : f32
|
|
// CHECK-DAG: %[[TWO:.+]] = arith.constant 2.000000e+00 : f32
|
|
// CHECK: %[[DOUBLEDX:.+]] = arith.mulf %arg0, %[[TWO]] : f32
|
|
// CHECK: %[[NEGDOUBLEDX:.+]] = arith.negf %[[DOUBLEDX]] : f32
|
|
// CHECK: %[[EXP1:.+]] = math.exp %[[NEGDOUBLEDX]] : f32
|
|
// CHECK: %[[DIVIDEND1:.+]] = arith.subf %[[ONE]], %[[EXP1]] : f32
|
|
// CHECK: %[[DIVISOR1:.+]] = arith.addf %[[EXP1]], %[[ONE]] : f32
|
|
// CHECK: %[[RES1:.+]] = arith.divf %[[DIVIDEND1]], %[[DIVISOR1]] : f32
|
|
// CHECK: %[[EXP2:.+]] = math.exp %[[DOUBLEDX]] : f32
|
|
// CHECK: %[[DIVIDEND2:.+]] = arith.subf %[[EXP2]], %[[ONE]] : f32
|
|
// CHECK: %[[DIVISOR2:.+]] = arith.addf %[[EXP2]], %[[ONE]] : f32
|
|
// CHECK: %[[RES2:.+]] = arith.divf %[[DIVIDEND2]], %[[DIVISOR2]] : f32
|
|
// CHECK: %[[COND:.+]] = arith.cmpf oge, %arg0, %[[ZERO]] : f32
|
|
// CHECK: %[[RESULT:.+]] = arith.select %[[COND]], %[[RES1]], %[[RES2]] : f32
|
|
// CHECK: return %[[RESULT]]
|