maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][Math] Add constant folder for TanOp. 

This patch adds constant folder for TanOp which only supports single and double precision floating-point.

Differential Revision: https://reviews.llvm.org/D130873
This commit is contained in:
jacquesguan 2022-08-01 15:52:55 +08:00
parent 39cfde2366
commit f1033a3f47
3 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mlir/include/mlir/Dialect/Math/IR/MathOps.td
View File

@ -656,6 +656,7 @@ def Math_TanOp : Math_FloatUnaryOp<"tan"> {
%a = math.tan %b : f64 %a = math.tan %b : f64
``` ```
}]; }];
let hasFolder = 1;
} }
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//

18
mlir/lib/Dialect/Math/IR/MathOps.cpp
View File

@ -264,6 +264,24 @@ OpFoldResult math::ExpM1Op::fold(ArrayRef<Attribute> operands) {
}); });
} }
//===----------------------------------------------------------------------===//
// TanOp folder
//===----------------------------------------------------------------------===//
OpFoldResult math::TanOp::fold(ArrayRef<Attribute> operands) {
return constFoldUnaryOpConditional<FloatAttr>(
operands, [](const APFloat &a) -> Optional<APFloat> {
switch (a.getSizeInBits(a.getSemantics())) {
case 64:
return APFloat(tan(a.convertToDouble()));
case 32:
return APFloat(tanf(a.convertToFloat()));
default:
return {};
}
});
}
/// Materialize an integer or floating point constant. /// Materialize an integer or floating point constant.
Operation *math::MathDialect::materializeConstant(OpBuilder &builder, Operation *math::MathDialect::materializeConstant(OpBuilder &builder,
Attribute value, Type type, Attribute value, Type type,

20
mlir/test/Dialect/Math/canonicalize.mlir
View File

@ -282,3 +282,23 @@ func.func @expm1_fold_vec() -> (vector<4xf32>) {
%0 = math.expm1 %v1 : vector<4xf32> %0 = math.expm1 %v1 : vector<4xf32>
return %0 : vector<4xf32> return %0 : vector<4xf32>
} }
// CHECK-LABEL: @tan_fold
// CHECK-NEXT: %[[cst:.+]] = arith.constant 1.55740774 : f32
// CHECK-NEXT: return %[[cst]]
func.func @tan_fold() -> f32 {
%c = arith.constant 1.0 : f32
%r = math.tan %c : f32
return %r : f32
}
// CHECK-LABEL: @tan_fold_vec
// CHECK-NEXT: %[[cst:.+]] = arith.constant dense<[0.000000e+00, 1.55740774, 0.000000e+00, 1.55740774]> : vector<4xf32>
// CHECK-NEXT: return %[[cst]]
func.func @tan_fold_vec() -> (vector<4xf32>) {
%v1 = arith.constant dense<[0.0, 1.0, 0.0, 1.0]> : vector<4xf32>
%0 = math.tan %v1 : vector<4xf32>
return %0 : vector<4xf32>
}