diff --git a/clang/lib/AST/Interp/Floating.h b/clang/lib/AST/Interp/Floating.h index 447361a43696b..f98f49a7afb70 100644 --- a/clang/lib/AST/Interp/Floating.h +++ b/clang/lib/AST/Interp/Floating.h @@ -87,6 +87,7 @@ class Floating final { bool isMin() const { return F.isSmallest(); } bool isMinusOne() const { return F.isExactlyValue(-1.0); } bool isNan() const { return F.isNaN(); } + bool isInf() const { return F.isInfinity(); } bool isFinite() const { return F.isFinite(); } ComparisonCategoryResult compare(const Floating &RHS) const { diff --git a/clang/lib/AST/Interp/InterpBuiltin.cpp b/clang/lib/AST/Interp/InterpBuiltin.cpp index 01c41339bba8c..b5e19af5c3728 100644 --- a/clang/lib/AST/Interp/InterpBuiltin.cpp +++ b/clang/lib/AST/Interp/InterpBuiltin.cpp @@ -173,6 +173,20 @@ static bool interp__builtin_isnan(InterpState &S, CodePtr OpPC, return true; } +static bool interp__builtin_isinf(InterpState &S, CodePtr OpPC, + const InterpFrame *Frame, const Function *F, + bool CheckSign) { + const Floating &Arg = S.Stk.peek(); + bool IsInf = Arg.isInf(); + + if (CheckSign) + S.Stk.push>( + Integral<32, true>::from(IsInf ? (Arg.isNegative() ? -1 : 1) : 0)); + else + S.Stk.push>(Integral<32, true>::from(Arg.isInf())); + return true; +} + bool InterpretBuiltin(InterpState &S, CodePtr OpPC, const Function *F) { InterpFrame *Frame = S.Current; APValue Dummy; @@ -239,6 +253,16 @@ bool InterpretBuiltin(InterpState &S, CodePtr OpPC, const Function *F) { return Ret(S, OpPC, Dummy); break; + case Builtin::BI__builtin_isinf: + if (interp__builtin_isinf(S, OpPC, Frame, F, /*Sign=*/false)) + return Ret(S, OpPC, Dummy); + break; + + case Builtin::BI__builtin_isinf_sign: + if (interp__builtin_isinf(S, OpPC, Frame, F, /*Sign=*/true)) + return Ret(S, OpPC, Dummy); + break; + default: return false; } diff --git a/clang/test/AST/Interp/builtin-functions.cpp b/clang/test/AST/Interp/builtin-functions.cpp index 8ff717217e472..7eb3e187d28ea 100644 --- a/clang/test/AST/Interp/builtin-functions.cpp +++ b/clang/test/AST/Interp/builtin-functions.cpp @@ -72,3 +72,8 @@ namespace fmin { constexpr float min3 = __builtin_fmin(__builtin_inf(), __builtin_nan("")); static_assert(min3 == __builtin_inf(), ""); } + +namespace inf { + static_assert(__builtin_isinf(__builtin_inf()), ""); + static_assert(!__builtin_isinf(1.0), ""); +}