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

[mlir][math] Lower math.floor,ceil to libm 

Lower math.floor and math.ceil to libm

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D131876
This commit is contained in:
Kai Sasaki 2022-08-24 10:58:01 +09:00 committed by Lewuathe
parent f250b97222
commit ad714d5b74
2 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/lib/Conversion/MathToLibm/MathToLibm.cpp
View File

@ -171,6 +171,10 @@ void mlir::populateMathToLibmConversionPatterns(
"sin", benefit);
patterns.add<ScalarOpToLibmCall<math::Log1pOp>>(
patterns.getContext(), "log1pf", "log1p", log1pBenefit.value_or(benefit));
patterns.add<ScalarOpToLibmCall<math::FloorOp>>(patterns.getContext(),
"floorf", "floor", benefit);
patterns.add<ScalarOpToLibmCall<math::CeilOp>>(patterns.getContext(), "ceilf",
"ceil", benefit);
}
namespace {

30
mlir/test/Conversion/MathToLibm/convert-to-libm.mlir
View File

@ -18,6 +18,10 @@
// CHECK-DAG: @cosf(f32) -> f32
// CHECK-DAG: @sin(f64) -> f64
// CHECK-DAG: @sinf(f32) -> f32
// CHECK-DAG: @floor(f64) -> f64
// CHECK-DAG: @floorf(f32) -> f32
// CHECK-DAG: @ceil(f64) -> f64
// CHECK-DAG: @ceilf(f32) -> f32
// CHECK-LABEL: func @atan_caller
// CHECK-SAME: %[[FLOAT:.*]]: f32
@ -314,4 +318,28 @@ func.func @log1p_caller(%float: f32, %double: f64) -> (f32, f64) {
%double_result = math.log1p %double : f64
// CHECK: return %[[FLOAT_RESULT]], %[[DOUBLE_RESULT]]
return %float_result, %double_result : f32, f64
}
}
// CHECK-LABEL: func @floor_caller
// CHECK-SAME: %[[FLOAT:.*]]: f32
// CHECK-SAME: %[[DOUBLE:.*]]: f64
func.func @floor_caller(%float: f32, %double: f64) -> (f32, f64) {
// CHECK-DAG: %[[FLOAT_RESULT:.*]] = call @floorf(%[[FLOAT]]) : (f32) -> f32
%float_result = math.floor %float : f32
// CHECK-DAG: %[[DOUBLE_RESULT:.*]] = call @floor(%[[DOUBLE]]) : (f64) -> f64
%double_result = math.floor %double : f64
// CHECK: return %[[FLOAT_RESULT]], %[[DOUBLE_RESULT]]
return %float_result, %double_result : f32, f64
}
// CHECK-LABEL: func @ceil_caller
// CHECK-SAME: %[[FLOAT:.*]]: f32
// CHECK-SAME: %[[DOUBLE:.*]]: f64
func.func @ceil_caller(%float: f32, %double: f64) -> (f32, f64) {
// CHECK-DAG: %[[FLOAT_RESULT:.*]] = call @ceilf(%[[FLOAT]]) : (f32) -> f32
%float_result = math.ceil %float : f32
// CHECK-DAG: %[[DOUBLE_RESULT:.*]] = call @ceil(%[[DOUBLE]]) : (f64) -> f64
%double_result = math.ceil %double : f64
// CHECK: return %[[FLOAT_RESULT]], %[[DOUBLE_RESULT]]
return %float_result, %double_result : f32, f64
}