diff --git a/paddle/cinn/ir/ir_printer.cc b/paddle/cinn/ir/ir_printer.cc index 93f7c3b9861b1..d82514cade9c5 100644 --- a/paddle/cinn/ir/ir_printer.cc +++ b/paddle/cinn/ir/ir_printer.cc @@ -14,7 +14,7 @@ #include "paddle/cinn/ir/ir_printer.h" #include -#include +#include #include #include #include @@ -175,7 +175,7 @@ void IrPrinter::Visit(const FloatImm *x) { } } else { float v = TruncateInfinity(x->value); - if (IsCloseEqualBoundValue(v)) std::fesetround(FE_TOWARDZERO); + if (IsCloseEqualBoundValue(v)) v = std::trunc(v); ss << std::setprecision(std::numeric_limits::max_digits10); ss << std::showpoint; ss << v; diff --git a/test/cpp/pir/cinn/replace_cross_block_reduction_test.cc b/test/cpp/pir/cinn/replace_cross_block_reduction_test.cc index 48b01f47a53d8..5d0d9d7c8b65b 100644 --- a/test/cpp/pir/cinn/replace_cross_block_reduction_test.cc +++ b/test/cpp/pir/cinn/replace_cross_block_reduction_test.cc @@ -155,7 +155,7 @@ TEST(CrossBlockReductionReplacer, RSLayout) { ScheduleBlock(B__reduce_init) { i0, i1 = axis.bind(i, j) - B__reduce_init[i0, i1] = -3.40282346e+38f + B__reduce_init[i0, i1] = -3.40282347e+38f } thread_bind[blockIdx.y] for (reduce_k, 0, 8) {