[Mlir-commits] [mlir] f745c91 - Revert "[MLIR][Math] Add support for f16 in the expansion of math.roundeven"
Alexander Shaposhnikov
llvmlistbot at llvm.org
Thu Aug 17 16:22:18 PDT 2023
Author: Alexander Shaposhnikov
Date: 2023-08-17T23:22:05Z
New Revision: f745c91f6171c53e062e9949623e7ba323894d71
URL: https://github.com/llvm/llvm-project/commit/f745c91f6171c53e062e9949623e7ba323894d71
DIFF: https://github.com/llvm/llvm-project/commit/f745c91f6171c53e062e9949623e7ba323894d71.diff
LOG: Revert "[MLIR][Math] Add support for f16 in the expansion of math.roundeven"
This reverts commit b96f6cf62902ca96ed5aa62d4e158292280284e1
since it has broken some Windows build bots
(see https://reviews.llvm.org/D157204).
Will recommit a fixed version later.
Added:
Modified:
mlir/include/mlir/ExecutionEngine/Float16bits.h
mlir/include/mlir/IR/BuiltinTypes.h
mlir/lib/Dialect/Math/Transforms/ExpandPatterns.cpp
mlir/test/CMakeLists.txt
mlir/test/Dialect/Math/expand-math.mlir
mlir/test/lit.cfg.py
mlir/test/mlir-cpu-runner/test-expand-math-approx.mlir
Removed:
################################################################################
diff --git a/mlir/include/mlir/ExecutionEngine/Float16bits.h b/mlir/include/mlir/ExecutionEngine/Float16bits.h
index 5eb1f2ce07639d..6bf1589aa13a2a 100644
--- a/mlir/include/mlir/ExecutionEngine/Float16bits.h
+++ b/mlir/include/mlir/ExecutionEngine/Float16bits.h
@@ -48,8 +48,5 @@ MLIR_FLOAT16_EXPORT std::ostream &operator<<(std::ostream &os, const f16 &f);
// Outputs a bfloat value.
MLIR_FLOAT16_EXPORT std::ostream &operator<<(std::ostream &os, const bf16 &d);
-extern "C" MLIR_FLOAT16_EXPORT void printF16(uint16_t bits);
-extern "C" MLIR_FLOAT16_EXPORT void printBF16(uint16_t bits);
-
#undef MLIR_FLOAT16_EXPORT
#endif // MLIR_EXECUTIONENGINE_FLOAT16BITS_H_
diff --git a/mlir/include/mlir/IR/BuiltinTypes.h b/mlir/include/mlir/IR/BuiltinTypes.h
index ce68fc2673dcaf..de363fc6370c2c 100644
--- a/mlir/include/mlir/IR/BuiltinTypes.h
+++ b/mlir/include/mlir/IR/BuiltinTypes.h
@@ -67,7 +67,6 @@ class FloatType : public Type {
unsigned getWidth();
/// Return the width of the mantissa of this type.
- /// The width includes the integer bit.
unsigned getFPMantissaWidth();
/// Get or create a new FloatType with bitwidth scaled by `scale`.
diff --git a/mlir/lib/Dialect/Math/Transforms/ExpandPatterns.cpp b/mlir/lib/Dialect/Math/Transforms/ExpandPatterns.cpp
index 93aff13cf5e2f2..ee8f23cf362b62 100644
--- a/mlir/lib/Dialect/Math/Transforms/ExpandPatterns.cpp
+++ b/mlir/lib/Dialect/Math/Transforms/ExpandPatterns.cpp
@@ -305,39 +305,31 @@ static LogicalResult convertRoundEvenOp(math::RoundEvenOp op,
Type operandETy = getElementTypeOrSelf(operandTy);
Type resultETy = getElementTypeOrSelf(resultTy);
- if (!isa<FloatType>(operandETy) || !isa<FloatType>(resultETy)) {
- return rewriter.notifyMatchFailure(op, "not a roundeven of f16 or f32.");
+ if (!operandETy.isF32() || !resultETy.isF32()) {
+ return rewriter.notifyMatchFailure(op, "not a roundeven of f32.");
}
- Type fTy = operandTy;
- Type iTy = rewriter.getIntegerType(operandETy.getIntOrFloatBitWidth());
- if (auto shapedTy = dyn_cast<ShapedType>(fTy)) {
- iTy = shapedTy.clone(iTy);
+ Type i32Ty = b.getI32Type();
+ Type f32Ty = b.getF32Type();
+ if (auto shapedTy = dyn_cast<ShapedType>(operandTy)) {
+ i32Ty = shapedTy.clone(i32Ty);
+ f32Ty = shapedTy.clone(f32Ty);
}
- unsigned bitWidth = operandETy.getIntOrFloatBitWidth();
- // The width returned by getFPMantissaWidth includes the integer bit.
- unsigned mantissaWidth =
- llvm::cast<FloatType>(operandETy).getFPMantissaWidth() - 1;
- unsigned exponentWidth = bitWidth - mantissaWidth - 1;
-
- // The names of the variables correspond to f32.
- // f32: 1 bit sign | 8 bits exponent | 23 bits mantissa.
- // f16: 1 bit sign | 5 bits exponent | 10 bits mantissa.
- Value c1Float = createFloatConst(loc, fTy, 1.0, b);
- Value c0 = createIntConst(loc, iTy, 0, b);
- Value c1 = createIntConst(loc, iTy, 1, b);
- Value cNeg1 = createIntConst(loc, iTy, -1, b);
- Value c23 = createIntConst(loc, iTy, mantissaWidth, b);
- Value c31 = createIntConst(loc, iTy, bitWidth - 1, b);
- Value c127 = createIntConst(loc, iTy, (1 << (exponentWidth - 1)) - 1, b);
- Value c2To22 = createIntConst(loc, iTy, 1 << (mantissaWidth - 1), b);
- Value c23Mask = createIntConst(loc, iTy, (1 << mantissaWidth) - 1, b);
- Value expMask = createIntConst(loc, iTy, (1 << exponentWidth) - 1, b);
-
- Value operandBitcast = b.create<arith::BitcastOp>(iTy, operand);
+ Value c1Float = createFloatConst(loc, f32Ty, 1.0, b);
+ Value c0 = createIntConst(loc, i32Ty, 0, b);
+ Value c1 = createIntConst(loc, i32Ty, 1, b);
+ Value cNeg1 = createIntConst(loc, i32Ty, -1, b);
+ Value c23 = createIntConst(loc, i32Ty, 23, b);
+ Value c31 = createIntConst(loc, i32Ty, 31, b);
+ Value c127 = createIntConst(loc, i32Ty, 127, b);
+ Value c2To22 = createIntConst(loc, i32Ty, 1 << 22, b);
+ Value c23Mask = createIntConst(loc, i32Ty, (1 << 23) - 1, b);
+ Value expMask = createIntConst(loc, i32Ty, (1 << 8) - 1, b);
+
+ Value operandBitcast = b.create<arith::BitcastOp>(i32Ty, operand);
Value round = b.create<math::RoundOp>(operand);
- Value roundBitcast = b.create<arith::BitcastOp>(iTy, round);
+ Value roundBitcast = b.create<arith::BitcastOp>(i32Ty, round);
// Get biased exponents for operand and round(operand)
Value operandExp = b.create<arith::AndIOp>(
@@ -348,7 +340,7 @@ static LogicalResult convertRoundEvenOp(math::RoundEvenOp op,
Value roundBiasedExp = b.create<arith::SubIOp>(roundExp, c127);
auto safeShiftRight = [&](Value x, Value shift) -> Value {
- // Clamp shift to valid range [0, bitwidth - 1] to avoid undefined behavior
+ // Clamp shift to valid range [0, 31] to avoid undefined behavior
Value clampedShift = b.create<arith::MaxSIOp>(shift, c0);
clampedShift = b.create<arith::MinSIOp>(clampedShift, c31);
return b.create<arith::ShRUIOp>(x, clampedShift);
diff --git a/mlir/test/CMakeLists.txt b/mlir/test/CMakeLists.txt
index 66a9cb01106ba5..b28c7cf140fa3b 100644
--- a/mlir/test/CMakeLists.txt
+++ b/mlir/test/CMakeLists.txt
@@ -119,7 +119,6 @@ if(LLVM_ENABLE_PIC AND TARGET ${LLVM_NATIVE_ARCH})
mlir-capi-execution-engine-test
mlir_c_runner_utils
mlir_runner_utils
- mlir_float16_utils
)
endif()
diff --git a/mlir/test/Dialect/Math/expand-math.mlir b/mlir/test/Dialect/Math/expand-math.mlir
index 9df7c4fd5ffb02..c28e2141db061a 100644
--- a/mlir/test/Dialect/Math/expand-math.mlir
+++ b/mlir/test/Dialect/Math/expand-math.mlir
@@ -232,8 +232,8 @@ func.func @powf_func(%a: f64, %b: f64) ->f64 {
// -----
-// CHECK-LABEL: func.func @roundeven32
-func.func @roundeven32(%arg: f32) -> f32 {
+// CHECK-LABEL: func.func @roundeven
+func.func @roundeven(%arg: f32) -> f32 {
%res = math.roundeven %arg : f32
return %res : f32
}
@@ -331,90 +331,3 @@ func.func @roundeven32(%arg: f32) -> f32 {
// CHECK: %[[COPYSIGN:.*]] = math.copysign %[[RESULT]], %[[VAL_0]] : f32
// CHECK: return %[[COPYSIGN]] : f32
-
-// -----
-
-// CHECK-LABEL: func.func @roundeven16
-func.func @roundeven16(%arg: f16) -> f16 {
- %res = math.roundeven %arg : f16
- return %res : f16
-}
-
-// CHECK-SAME: %[[VAL_0:.*]]: f16) -> f16 {
-// CHECK-DAG: %[[C_0:.*]] = arith.constant 0 : i16
-// CHECK-DAG: %[[C_1:.*]] = arith.constant 1 : i16
-// CHECK-DAG: %[[C_NEG_1:.*]] = arith.constant -1 : i16
-// CHECK-DAG: %[[C_1_FLOAT:.*]] = arith.constant 1.000000e+00 : f16
-// CHECK-DAG: %[[C_10:.*]] = arith.constant 10 : i16
-// CHECK-DAG: %[[C_15:.*]] = arith.constant 15 : i16
-// CHECK-DAG: %[[C_512:.*]] = arith.constant 512 : i16
-// CHECK-DAG: %[[C_1023:.*]] = arith.constant 1023 : i16
-// CHECK-DAG: %[[EXP_MASK:.*]] = arith.constant 31 : i16
-
-// CHECK: %[[OPERAND_BITCAST:.*]] = arith.bitcast %[[VAL_0]] : f16 to i16
-// CHECK: %[[ROUND:.*]] = math.round %[[VAL_0]] : f16
-// CHECK: %[[ROUND_BITCAST:.*]] = arith.bitcast %[[ROUND]] : f16 to i16
-
-// Get biased exponents of `round` and `operand`
-// CHECK: %[[SHIFTED_OPERAND_BITCAST:.*]] = arith.shrui %[[OPERAND_BITCAST]], %[[C_10]] : i16
-// CHECK: %[[OPERAND_EXP:.*]] = arith.andi %[[SHIFTED_OPERAND_BITCAST]], %[[EXP_MASK]] : i16
-// CHECK: %[[OPERAND_BIASED_EXP:.*]] = arith.subi %[[OPERAND_EXP]], %[[C_15]] : i16
-// CHECK: %[[SHIFTED_ROUND_BITCAST:.*]] = arith.shrui %[[ROUND_BITCAST]], %[[C_10]] : i16
-// CHECK: %[[ROUND_EXP:.*]] = arith.andi %[[SHIFTED_ROUND_BITCAST]], %[[EXP_MASK]] : i16
-// CHECK: %[[ROUND_BIASED_EXP:.*]] = arith.subi %[[ROUND_EXP]], %[[C_15]] : i16
-
-// Determine if `ROUND_BITCAST` is an even whole number or a special value
-// +-inf, +-nan.
-// Mask mantissa of `ROUND_BITCAST` with a mask shifted to the right by
-// `ROUND_BIASED_EXP - 1`
-// CHECK-DAG: %[[ROUND_BIASED_EXP_MINUS_1:.*]] = arith.subi %[[ROUND_BIASED_EXP]], %[[C_1]] : i16
-// CHECK-DAG: %[[CLAMPED_SHIFT_0:.*]] = arith.maxsi %[[ROUND_BIASED_EXP_MINUS_1]], %[[C_0]] : i16
-// CHECK-DAG: %[[CLAMPED_SHIFT_1:.*]] = arith.minsi %[[CLAMPED_SHIFT_0]], %[[C_15]] : i16
-// CHECK-DAG: %[[SHIFTED_MANTISSA_MASK_0:.*]] = arith.shrui %[[C_1023]], %[[CLAMPED_SHIFT_1]] : i16
-// CHECK-DAG: %[[ROUND_MASKED_MANTISSA:.*]] = arith.andi %[[ROUND_BITCAST]], %[[SHIFTED_MANTISSA_MASK_0]] : i16
-
-// `ROUND_BITCAST` is not even whole number or special value if masked
-// mantissa is != 0 or `ROUND_BIASED_EXP == 0`
-// CHECK-DAG: %[[ROUND_IS_NOT_EVEN_OR_SPECIAL_0:.*]] = arith.cmpi ne, %[[ROUND_MASKED_MANTISSA]], %[[C_0]] : i16
-// CHECK-DAG: %[[ROUND_BIASED_EXP_EQ_0:.*]] = arith.cmpi eq, %[[ROUND_BIASED_EXP]], %[[C_0]] : i16
-// CHECK-DAG: %[[ROUND_IS_NOT_EVEN_OR_SPECIAL_1:.*]] = arith.ori %[[ROUND_IS_NOT_EVEN_OR_SPECIAL_0]], %[[ROUND_BIASED_EXP_EQ_0]] : i1
-
-// Determine if operand is halfway between two integer values
-// CHECK: %[[OPERAND_BIASED_EXP_EQ_NEG_1:.*]] = arith.cmpi eq, %[[OPERAND_BIASED_EXP]], %[[C_NEG_1]] : i16
-// CHECK: %[[CLAMPED_SHIFT_2:.*]] = arith.maxsi %[[OPERAND_BIASED_EXP]], %[[C_0]] : i16
-// CHECK: %[[CLAMPED_SHIFT_3:.*]] = arith.minsi %[[CLAMPED_SHIFT_2]], %[[C_15]] : i16
-// CHECK: %[[SHIFTED_2_TO_9:.*]] = arith.shrui %[[C_512]], %[[CLAMPED_SHIFT_3]] : i16
-
-// A value with `0 <= BIASED_EXP < 10` is halfway between two consecutive
-// integers if the bit at index `BIASED_EXP` starting from the left in the
-// mantissa is 1 and all the bits to the right are zero. For the case where
-// `BIASED_EXP == -1, the expected mantissa is all zeros.
-// CHECK: %[[EXPECTED_OPERAND_MASKED_MANTISSA:.*]] = arith.select %[[OPERAND_BIASED_EXP_EQ_NEG_1]], %[[C_0]], %[[SHIFTED_2_TO_9]] : i16
-
-// Mask mantissa of `OPERAND_BITCAST` with a mask shifted to the right by
-// `OPERAND_BIASED_EXP`
-// CHECK: %[[CLAMPED_SHIFT_4:.*]] = arith.maxsi %[[OPERAND_BIASED_EXP]], %[[C_0]] : i16
-// CHECK: %[[CLAMPED_SHIFT_5:.*]] = arith.minsi %[[CLAMPED_SHIFT_4]], %[[C_15]] : i16
-// CHECK: %[[SHIFTED_MANTISSA_MASK_1:.*]] = arith.shrui %[[C_1023]], %[[CLAMPED_SHIFT_5]] : i16
-// CHECK: %[[OPERAND_MASKED_MANTISSA:.*]] = arith.andi %[[OPERAND_BITCAST]], %[[SHIFTED_MANTISSA_MASK_1]] : i16
-
-// The operand is halfway between two integers if the masked mantissa is equal
-// to the expected mantissa and the biased exponent is in the range
-// [-1, 23).
-// CHECK-DAG: %[[OPERAND_BIASED_EXP_GE_NEG_1:.*]] = arith.cmpi sge, %[[OPERAND_BIASED_EXP]], %[[C_NEG_1]] : i16
-// CHECK-DAG: %[[OPERAND_BIASED_EXP_LT_10:.*]] = arith.cmpi slt, %[[OPERAND_BIASED_EXP]], %[[C_10]] : i16
-// CHECK-DAG: %[[OPERAND_IS_HALFWAY_0:.*]] = arith.cmpi eq, %[[OPERAND_MASKED_MANTISSA]], %[[EXPECTED_OPERAND_MASKED_MANTISSA]] : i16
-// CHECK-DAG: %[[OPERAND_IS_HALFWAY_1:.*]] = arith.andi %[[OPERAND_IS_HALFWAY_0]], %[[OPERAND_BIASED_EXP_LT_10]] : i1
-// CHECK-DAG: %[[OPERAND_IS_HALFWAY_2:.*]] = arith.andi %[[OPERAND_IS_HALFWAY_1]], %[[OPERAND_BIASED_EXP_GE_NEG_1]] : i1
-
-// Adjust rounded operand with `round(operand) - sign(operand)` to correct the
-// case where `round` rounded in the oppositve direction of `roundeven`.
-// CHECK: %[[SIGN:.*]] = math.copysign %[[C_1_FLOAT]], %[[VAL_0]] : f16
-// CHECK: %[[ROUND_SHIFTED:.*]] = arith.subf %[[ROUND]], %[[SIGN]] : f16
-// CHECK: %[[NEEDS_SHIFT:.*]] = arith.andi %[[ROUND_IS_NOT_EVEN_OR_SPECIAL_1]], %[[OPERAND_IS_HALFWAY_2]] : i1
-// CHECK: %[[RESULT:.*]] = arith.select %[[NEEDS_SHIFT]], %[[ROUND_SHIFTED]], %[[ROUND]] : f16
-
-// The `x - sign` adjustment does not preserve the sign when we are adjusting the value -1 to -0.
-// CHECK: %[[COPYSIGN:.*]] = math.copysign %[[RESULT]], %[[VAL_0]] : f16
-
-// CHECK: return %[[COPYSIGN]] : f16
diff --git a/mlir/test/lit.cfg.py b/mlir/test/lit.cfg.py
index f265ac794c6f6d..fb99422c3ff5a3 100644
--- a/mlir/test/lit.cfg.py
+++ b/mlir/test/lit.cfg.py
@@ -98,7 +98,6 @@ def add_runtime(name):
add_runtime("mlir_runner_utils"),
add_runtime("mlir_c_runner_utils"),
add_runtime("mlir_async_runtime"),
- add_runtime("mlir_float16_utils"),
"mlir-linalg-ods-yaml-gen",
"mlir-reduce",
"mlir-pdll",
diff --git a/mlir/test/mlir-cpu-runner/test-expand-math-approx.mlir b/mlir/test/mlir-cpu-runner/test-expand-math-approx.mlir
index 101cb6b065edeb..d6943e5fc2831d 100644
--- a/mlir/test/mlir-cpu-runner/test-expand-math-approx.mlir
+++ b/mlir/test/mlir-cpu-runner/test-expand-math-approx.mlir
@@ -3,7 +3,6 @@
// RUN: -e main -entry-point-result=void -O0 \
// RUN: -shared-libs=%mlir_c_runner_utils \
// RUN: -shared-libs=%mlir_runner_utils \
-// RUN: -shared-libs=%mlir_float16_utils \
// RUN: | FileCheck %s
// -------------------------------------------------------------------------- //
@@ -244,26 +243,26 @@ func.func @powf() {
// roundeven.
// -------------------------------------------------------------------------- //
-func.func @func_roundeven32(%a : f32) {
+func.func @func_roundeven(%a : f32) {
%b = math.roundeven %a : f32
vector.print %b : f32
return
}
-func.func @func_roundeven32$bitcast_result_to_int(%a : f32) {
+func.func @func_roundeven$bitcast_result_to_int(%a : f32) {
%b = math.roundeven %a : f32
%c = arith.bitcast %b : f32 to i32
vector.print %c : i32
return
}
-func.func @func_roundeven32$vector(%a : vector<1xf32>) {
+func.func @func_roundeven$vector(%a : vector<1xf32>) {
%b = math.roundeven %a : vector<1xf32>
vector.print %b : vector<1xf32>
return
}
-func.func @roundeven32() {
+func.func @roundeven() {
%c0_25 = arith.constant 0.25 : f32
%c0_5 = arith.constant 0.5 : f32
%c0_75 = arith.constant 0.75 : f32
@@ -297,66 +296,66 @@ func.func @roundeven32() {
%cNeg3_75 = arith.constant -3.75 : f32
// CHECK-NEXT: 0
- call @func_roundeven32(%c0_25) : (f32) -> ()
+ call @func_roundeven(%c0_25) : (f32) -> ()
// CHECK-NEXT: 0
- call @func_roundeven32(%c0_5) : (f32) -> ()
+ call @func_roundeven(%c0_5) : (f32) -> ()
// CHECK-NEXT: 1
- call @func_roundeven32(%c0_75) : (f32) -> ()
+ call @func_roundeven(%c0_75) : (f32) -> ()
// CHECK-NEXT: 1
- call @func_roundeven32(%c1) : (f32) -> ()
+ call @func_roundeven(%c1) : (f32) -> ()
// CHECK-NEXT: 1
- call @func_roundeven32(%c1_25) : (f32) -> ()
+ call @func_roundeven(%c1_25) : (f32) -> ()
// CHECK-NEXT: 2
- call @func_roundeven32(%c1_5) : (f32) -> ()
+ call @func_roundeven(%c1_5) : (f32) -> ()
// CHECK-NEXT: 2
- call @func_roundeven32(%c1_75) : (f32) -> ()
+ call @func_roundeven(%c1_75) : (f32) -> ()
// CHECK-NEXT: 2
- call @func_roundeven32(%c2) : (f32) -> ()
+ call @func_roundeven(%c2) : (f32) -> ()
// CHECK-NEXT: 2
- call @func_roundeven32(%c2_25) : (f32) -> ()
+ call @func_roundeven(%c2_25) : (f32) -> ()
// CHECK-NEXT: 2
- call @func_roundeven32(%c2_5) : (f32) -> ()
+ call @func_roundeven(%c2_5) : (f32) -> ()
// CHECK-NEXT: 3
- call @func_roundeven32(%c2_75) : (f32) -> ()
+ call @func_roundeven(%c2_75) : (f32) -> ()
// CHECK-NEXT: 3
- call @func_roundeven32(%c3) : (f32) -> ()
+ call @func_roundeven(%c3) : (f32) -> ()
// CHECK-NEXT: 3
- call @func_roundeven32(%c3_25) : (f32) -> ()
+ call @func_roundeven(%c3_25) : (f32) -> ()
// CHECK-NEXT: 4
- call @func_roundeven32(%c3_5) : (f32) -> ()
+ call @func_roundeven(%c3_5) : (f32) -> ()
// CHECK-NEXT: 4
- call @func_roundeven32(%c3_75) : (f32) -> ()
+ call @func_roundeven(%c3_75) : (f32) -> ()
// CHECK-NEXT: -0
- call @func_roundeven32(%cNeg0_25) : (f32) -> ()
+ call @func_roundeven(%cNeg0_25) : (f32) -> ()
// CHECK-NEXT: -0
- call @func_roundeven32(%cNeg0_5) : (f32) -> ()
+ call @func_roundeven(%cNeg0_5) : (f32) -> ()
// CHECK-NEXT: -1
- call @func_roundeven32(%cNeg0_75) : (f32) -> ()
+ call @func_roundeven(%cNeg0_75) : (f32) -> ()
// CHECK-NEXT: -1
- call @func_roundeven32(%cNeg1) : (f32) -> ()
+ call @func_roundeven(%cNeg1) : (f32) -> ()
// CHECK-NEXT: -1
- call @func_roundeven32(%cNeg1_25) : (f32) -> ()
+ call @func_roundeven(%cNeg1_25) : (f32) -> ()
// CHECK-NEXT: -2
- call @func_roundeven32(%cNeg1_5) : (f32) -> ()
+ call @func_roundeven(%cNeg1_5) : (f32) -> ()
// CHECK-NEXT: -2
- call @func_roundeven32(%cNeg1_75) : (f32) -> ()
+ call @func_roundeven(%cNeg1_75) : (f32) -> ()
// CHECK-NEXT: -2
- call @func_roundeven32(%cNeg2) : (f32) -> ()
+ call @func_roundeven(%cNeg2) : (f32) -> ()
// CHECK-NEXT: -2
- call @func_roundeven32(%cNeg2_25) : (f32) -> ()
+ call @func_roundeven(%cNeg2_25) : (f32) -> ()
// CHECK-NEXT: -2
- call @func_roundeven32(%cNeg2_5) : (f32) -> ()
+ call @func_roundeven(%cNeg2_5) : (f32) -> ()
// CHECK-NEXT: -3
- call @func_roundeven32(%cNeg2_75) : (f32) -> ()
+ call @func_roundeven(%cNeg2_75) : (f32) -> ()
// CHECK-NEXT: -3
- call @func_roundeven32(%cNeg3) : (f32) -> ()
+ call @func_roundeven(%cNeg3) : (f32) -> ()
// CHECK-NEXT: -3
- call @func_roundeven32(%cNeg3_25) : (f32) -> ()
+ call @func_roundeven(%cNeg3_25) : (f32) -> ()
// CHECK-NEXT: -4
- call @func_roundeven32(%cNeg3_5) : (f32) -> ()
+ call @func_roundeven(%cNeg3_5) : (f32) -> ()
// CHECK-NEXT: -4
- call @func_roundeven32(%cNeg3_75) : (f32) -> ()
+ call @func_roundeven(%cNeg3_75) : (f32) -> ()
// Special values: 0, -0, inf, -inf, nan, -nan
@@ -372,22 +371,22 @@ func.func @roundeven32() {
%cNegNan = arith.bitcast %cNegNanInt : i32 to f32
// CHECK-NEXT: -0
- call @func_roundeven32(%cNeg0) : (f32) -> ()
+ call @func_roundeven(%cNeg0) : (f32) -> ()
// CHECK-NEXT: 0
- call @func_roundeven32(%c0) : (f32) -> ()
+ call @func_roundeven(%c0) : (f32) -> ()
// CHECK-NEXT: inf
- call @func_roundeven32(%cInf) : (f32) -> ()
+ call @func_roundeven(%cInf) : (f32) -> ()
// CHECK-NEXT: -inf
- call @func_roundeven32(%cNegInf) : (f32) -> ()
+ call @func_roundeven(%cNegInf) : (f32) -> ()
// Per IEEE 754-2008, sign is not required when printing a negative NaN, so
// print as an int to ensure input NaN is left unchanged.
// CHECK-NEXT: 2143289344
// CHECK-NEXT: 2143289344
- call @func_roundeven32$bitcast_result_to_int(%cNan) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%cNan) : (f32) -> ()
vector.print %cNanInt : i32
// CHECK-NEXT: -4194304
// CHECK-NEXT: -4194304
- call @func_roundeven32$bitcast_result_to_int(%cNegNan) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%cNegNan) : (f32) -> ()
vector.print %cNegNanInt : i32
@@ -403,206 +402,25 @@ func.func @roundeven32() {
// CHECK-NEXT: 1258291196
// hex: 0x4AFFFFFC
- call @func_roundeven32$bitcast_result_to_int(%c8388606_5) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%c8388606_5) : (f32) -> ()
// CHECK-NEXT: 1258291198
// hex: 0x4AFFFFFE
- call @func_roundeven32$bitcast_result_to_int(%c8388607) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%c8388607) : (f32) -> ()
// CHECK-NEXT: 1258291200
// hex: 0x4B000000
- call @func_roundeven32$bitcast_result_to_int(%c8388607_5) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%c8388607_5) : (f32) -> ()
// CHECK-NEXT: 1258291200
// hex: 0x4B000000
- call @func_roundeven32$bitcast_result_to_int(%c8388608) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%c8388608) : (f32) -> ()
// CHECK-NEXT: 1258291201
// hex: 0x4B000001
- call @func_roundeven32$bitcast_result_to_int(%c8388609) : (f32) -> ()
+ call @func_roundeven$bitcast_result_to_int(%c8388609) : (f32) -> ()
// Check that vector type works
%cVec = arith.constant dense<[0.5]> : vector<1xf32>
// CHECK-NEXT: ( 0 )
- call @func_roundeven32$vector(%cVec) : (vector<1xf32>) -> ()
- return
-}
-
-func.func @func_roundeven16(%a : f16) {
- %b = math.roundeven %a : f16
- vector.print %b : f16
- return
-}
-
-func.func @func_roundeven16$bitcast_result_to_int(%a : f16) {
- %b = math.roundeven %a : f16
- %c = arith.bitcast %b : f16 to i16
- vector.print %c : i16
- return
-}
-
-func.func @func_roundeven16$vector(%a : vector<1xf16>) {
- %b = math.roundeven %a : vector<1xf16>
- vector.print %b : vector<1xf16>
- return
-}
-
-func.func @roundeven16() {
- %c0_25 = arith.constant 0.25 : f16
- %c0_5 = arith.constant 0.5 : f16
- %c0_75 = arith.constant 0.75 : f16
- %c1 = arith.constant 1.0 : f16
- %c1_25 = arith.constant 1.25 : f16
- %c1_5 = arith.constant 1.5 : f16
- %c1_75 = arith.constant 1.75 : f16
- %c2 = arith.constant 2.0 : f16
- %c2_25 = arith.constant 2.25 : f16
- %c2_5 = arith.constant 2.5 : f16
- %c2_75 = arith.constant 2.75 : f16
- %c3 = arith.constant 3.0 : f16
- %c3_25 = arith.constant 3.25 : f16
- %c3_5 = arith.constant 3.5 : f16
- %c3_75 = arith.constant 3.75 : f16
-
- %cNeg0_25 = arith.constant -0.25 : f16
- %cNeg0_5 = arith.constant -0.5 : f16
- %cNeg0_75 = arith.constant -0.75 : f16
- %cNeg1 = arith.constant -1.0 : f16
- %cNeg1_25 = arith.constant -1.25 : f16
- %cNeg1_5 = arith.constant -1.5 : f16
- %cNeg1_75 = arith.constant -1.75 : f16
- %cNeg2 = arith.constant -2.0 : f16
- %cNeg2_25 = arith.constant -2.25 : f16
- %cNeg2_5 = arith.constant -2.5 : f16
- %cNeg2_75 = arith.constant -2.75 : f16
- %cNeg3 = arith.constant -3.0 : f16
- %cNeg3_25 = arith.constant -3.25 : f16
- %cNeg3_5 = arith.constant -3.5 : f16
- %cNeg3_75 = arith.constant -3.75 : f16
-
- // CHECK-NEXT: 0
- call @func_roundeven16(%c0_25) : (f16) -> ()
- // CHECK-NEXT: 0
- call @func_roundeven16(%c0_5) : (f16) -> ()
- // CHECK-NEXT: 1
- call @func_roundeven16(%c0_75) : (f16) -> ()
- // CHECK-NEXT: 1
- call @func_roundeven16(%c1) : (f16) -> ()
- // CHECK-NEXT: 1
- call @func_roundeven16(%c1_25) : (f16) -> ()
- // CHECK-NEXT: 2
- call @func_roundeven16(%c1_5) : (f16) -> ()
- // CHECK-NEXT: 2
- call @func_roundeven16(%c1_75) : (f16) -> ()
- // CHECK-NEXT: 2
- call @func_roundeven16(%c2) : (f16) -> ()
- // CHECK-NEXT: 2
- call @func_roundeven16(%c2_25) : (f16) -> ()
- // CHECK-NEXT: 2
- call @func_roundeven16(%c2_5) : (f16) -> ()
- // CHECK-NEXT: 3
- call @func_roundeven16(%c2_75) : (f16) -> ()
- // CHECK-NEXT: 3
- call @func_roundeven16(%c3) : (f16) -> ()
- // CHECK-NEXT: 3
- call @func_roundeven16(%c3_25) : (f16) -> ()
- // CHECK-NEXT: 4
- call @func_roundeven16(%c3_5) : (f16) -> ()
- // CHECK-NEXT: 4
- call @func_roundeven16(%c3_75) : (f16) -> ()
-
- // CHECK-NEXT: -0
- call @func_roundeven16(%cNeg0_25) : (f16) -> ()
- // CHECK-NEXT: -0
- call @func_roundeven16(%cNeg0_5) : (f16) -> ()
- // CHECK-NEXT: -1
- call @func_roundeven16(%cNeg0_75) : (f16) -> ()
- // CHECK-NEXT: -1
- call @func_roundeven16(%cNeg1) : (f16) -> ()
- // CHECK-NEXT: -1
- call @func_roundeven16(%cNeg1_25) : (f16) -> ()
- // CHECK-NEXT: -2
- call @func_roundeven16(%cNeg1_5) : (f16) -> ()
- // CHECK-NEXT: -2
- call @func_roundeven16(%cNeg1_75) : (f16) -> ()
- // CHECK-NEXT: -2
- call @func_roundeven16(%cNeg2) : (f16) -> ()
- // CHECK-NEXT: -2
- call @func_roundeven16(%cNeg2_25) : (f16) -> ()
- // CHECK-NEXT: -2
- call @func_roundeven16(%cNeg2_5) : (f16) -> ()
- // CHECK-NEXT: -3
- call @func_roundeven16(%cNeg2_75) : (f16) -> ()
- // CHECK-NEXT: -3
- call @func_roundeven16(%cNeg3) : (f16) -> ()
- // CHECK-NEXT: -3
- call @func_roundeven16(%cNeg3_25) : (f16) -> ()
- // CHECK-NEXT: -4
- call @func_roundeven16(%cNeg3_5) : (f16) -> ()
- // CHECK-NEXT: -4
- call @func_roundeven16(%cNeg3_75) : (f16) -> ()
-
- // Special values: 0, -0, inf, -inf, nan, -nan
- %cNeg0 = arith.constant -0.0 : f16
- %c0 = arith.constant 0.0 : f16
- %cInfInt = arith.constant 0x7c00 : i16
- %cInf = arith.bitcast %cInfInt : i16 to f16
- %cNegInfInt = arith.constant 0xfc00 : i16
- %cNegInf = arith.bitcast %cNegInfInt : i16 to f16
- %cNanInt = arith.constant 0x7e00 : i16
- %cNan = arith.bitcast %cNanInt : i16 to f16
- %cNegNanInt = arith.constant 0xfe00 : i16
- %cNegNan = arith.bitcast %cNegNanInt : i16 to f16
-
- // CHECK-NEXT: -0
- call @func_roundeven16(%cNeg0) : (f16) -> ()
- // CHECK-NEXT: 0
- call @func_roundeven16(%c0) : (f16) -> ()
- // CHECK-NEXT: inf
- call @func_roundeven16(%cInf) : (f16) -> ()
- // CHECK-NEXT: -inf
- call @func_roundeven16(%cNegInf) : (f16) -> ()
- // Per IEEE 754-2008, sign is not required when printing a negative NaN, so
- // print as an int to ensure input NaN is left unchanged.
- // CHECK-NEXT: 32256
- // CHECK-NEXT: 32256
- call @func_roundeven16$bitcast_result_to_int(%cNan) : (f16) -> ()
- vector.print %cNanInt : i16
- // CHECK-NEXT: -512
- // CHECK-NEXT: -512
- call @func_roundeven16$bitcast_result_to_int(%cNegNan) : (f16) -> ()
- vector.print %cNegNanInt : i16
-
- // Values above and below 2^10 = 1024
- %c1022_5 = arith.constant 1022.5 : f16
- %c1023 = arith.constant 1023.0 : f16
- %c1023_5 = arith.constant 1023.5 : f16
- %c1024 = arith.constant 1024.0 : f16
- %c1025 = arith.constant 1025.0 : f16
-
- // CHECK-NEXT: 25596
- // hex: 0x63fc
- call @func_roundeven16$bitcast_result_to_int(%c1022_5) : (f16) -> ()
- // CHECK-NEXT: 25598
- // hex: 0x63fe
- call @func_roundeven16$bitcast_result_to_int(%c1023) : (f16) -> ()
- // CHECK-NEXT: 25600
- // hex: 0x6400
- call @func_roundeven16$bitcast_result_to_int(%c1023_5) : (f16) -> ()
- // CHECK-NEXT: 25600
- // hex: 0x6400
- call @func_roundeven16$bitcast_result_to_int(%c1024) : (f16) -> ()
- // CHECK-NEXT: 25601
- // hex: 0x6401
- call @func_roundeven16$bitcast_result_to_int(%c1025) : (f16) -> ()
-
- // Check that vector type works
- %cVec = arith.constant dense<[0.5]> : vector<1xf16>
- // CHECK-NEXT: ( 0 )
- call @func_roundeven16$vector(%cVec) : (vector<1xf16>) -> ()
- return
-}
-
-func.func @roundeven() {
- call @roundeven32() : () -> ()
- call @roundeven16() : () -> ()
+ call @func_roundeven$vector(%cVec) : (vector<1xf32>) -> ()
return
}
More information about the Mlir-commits
mailing list