diff --git a/clang/include/clang/Sema/Sema.h b/clang/include/clang/Sema/Sema.h --- a/clang/include/clang/Sema/Sema.h +++ b/clang/include/clang/Sema/Sema.h @@ -710,6 +710,12 @@ return result; } + void resetFPOptions(FPOptions FPO) { + CurFPFeatures = FPO; + FpPragmaStack.Stack.clear(); + FpPragmaStack.CurrentValue = FPO.getChangesFrom(FPOptions(LangOpts)); + } + // RAII object to push / pop sentinel slots for all MS #pragma stacks. // Actions should be performed only if we enter / exit a C++ method body. class PragmaStackSentinelRAII { @@ -13974,6 +13980,8 @@ CachedTokens Toks; /// The template function declaration to be late parsed. Decl *D; + /// Floating-point options in the point of definition. + FPOptions FPO; }; template <> diff --git a/clang/lib/Parse/ParseTemplate.cpp b/clang/lib/Parse/ParseTemplate.cpp --- a/clang/lib/Parse/ParseTemplate.cpp +++ b/clang/lib/Parse/ParseTemplate.cpp @@ -1731,6 +1731,10 @@ Actions.PushDeclContext(Actions.getCurScope(), DC); } + // Parsing should occur with empty FP pragma stack and FP options used in the + // point of the template definition. + Actions.resetFPOptions(LPT.FPO); + assert(!LPT.Toks.empty() && "Empty body!"); // Append the current token at the end of the new token stream so that it diff --git a/clang/lib/Sema/SemaTemplate.cpp b/clang/lib/Sema/SemaTemplate.cpp --- a/clang/lib/Sema/SemaTemplate.cpp +++ b/clang/lib/Sema/SemaTemplate.cpp @@ -11342,6 +11342,7 @@ // Take tokens to avoid allocations LPT->Toks.swap(Toks); LPT->D = FnD; + LPT->FPO = getCurFPFeatures(); LateParsedTemplateMap.insert(std::make_pair(FD, std::move(LPT))); FD->setLateTemplateParsed(true); diff --git a/clang/test/CodeGen/fp-template.cpp b/clang/test/CodeGen/fp-template.cpp --- a/clang/test/CodeGen/fp-template.cpp +++ b/clang/test/CodeGen/fp-template.cpp @@ -15,4 +15,40 @@ // CHECK-SAME: (float noundef %{{.*}}, float noundef %{{.*}}) #[[ATTR01:[0-9]+]]{{.*}} { // CHECK: call float @llvm.experimental.constrained.fadd.f32 + +template +Ty templ_02(Ty x, Ty y) { + return x + y; +} + +#pragma STDC FENV_ROUND FE_UPWARD + +template +Ty templ_03(Ty x, Ty y) { + return x - y; +} + +#pragma STDC FENV_ROUND FE_TONEAREST + +float func_02(float x, float y) { + return templ_02(x, y); +} + +// CHECK-LABEL: define {{.*}} float @_Z8templ_02IfET_S0_S0_ +// CHECK: %add = fadd float %0, %1 + +float func_03(float x, float y) { + return templ_03(x, y); +} + +// CHECK-LABEL: define {{.*}} float @_Z8templ_03IfET_S0_S0_ +// CHECK: call float @llvm.experimental.constrained.fsub.f32({{.*}}, metadata !"round.upward", metadata !"fpexcept.ignore") + + +// This pragma sets non-default rounding mode before delayed parsing occurs. It +// is used to check that the parsing uses FP options defined by command line +// options or by pragma before the template definition but not by this pragma. +#pragma STDC FENV_ROUND FE_TOWARDZERO + + // CHECK: attributes #[[ATTR01]] = { {{.*}}strictfp