diff --git a/clang/include/clang/AST/ExprCXX.h b/clang/include/clang/AST/ExprCXX.h index 57ab94bcb2010..696a574833dad 100644 --- a/clang/include/clang/AST/ExprCXX.h +++ b/clang/include/clang/AST/ExprCXX.h @@ -4326,6 +4326,8 @@ class SizeOfPackExpr final /// Retrieve the parameter pack. NamedDecl *getPack() const { return Pack; } + void setPack(NamedDecl *NewPack) { Pack = NewPack; } + /// Retrieve the length of the parameter pack. /// /// This routine may only be invoked when the expression is not diff --git a/clang/lib/Sema/SemaTemplateInstantiate.cpp b/clang/lib/Sema/SemaTemplateInstantiate.cpp index de0ec0128905f..9c98cf11e43dd 100644 --- a/clang/lib/Sema/SemaTemplateInstantiate.cpp +++ b/clang/lib/Sema/SemaTemplateInstantiate.cpp @@ -1731,31 +1731,21 @@ namespace { return inherited::TransformLambdaBody(E, Body); } - ExprResult RebuildSizeOfPackExpr(SourceLocation OperatorLoc, - NamedDecl *Pack, SourceLocation PackLoc, - SourceLocation RParenLoc, - std::optional Length, - ArrayRef PartialArgs) { - if (SemaRef.CodeSynthesisContexts.back().Kind != - Sema::CodeSynthesisContext::ConstraintNormalization) - return inherited::RebuildSizeOfPackExpr(OperatorLoc, Pack, PackLoc, - RParenLoc, Length, PartialArgs); - -#ifndef NDEBUG - for (auto *Iter = TemplateArgs.begin(); Iter != TemplateArgs.end(); - ++Iter) - for (const TemplateArgument &TA : Iter->Args) - assert(TA.getKind() != TemplateArgument::Pack || TA.pack_size() == 1); -#endif - Sema::ArgumentPackSubstitutionIndexRAII SubstIndex( - SemaRef, /*NewSubstitutionIndex=*/0); - Decl *NewPack = TransformDecl(PackLoc, Pack); - if (!NewPack) - return ExprError(); - - return inherited::RebuildSizeOfPackExpr(OperatorLoc, - cast(NewPack), PackLoc, - RParenLoc, Length, PartialArgs); + ExprResult TransformSizeOfPackExpr(SizeOfPackExpr *E) { + ExprResult Transformed = inherited::TransformSizeOfPackExpr(E); + if (!Transformed.isUsable()) + return Transformed; + auto *TransformedExpr = cast(Transformed.get()); + if (SemaRef.CodeSynthesisContexts.back().Kind == + Sema::CodeSynthesisContext::ConstraintNormalization && + TransformedExpr->getPack() == E->getPack()) { + Decl *NewPack = + TransformDecl(E->getPackLoc(), TransformedExpr->getPack()); + if (!NewPack) + return ExprError(); + TransformedExpr->setPack(cast(NewPack)); + } + return TransformedExpr; } ExprResult TransformRequiresExpr(RequiresExpr *E) { @@ -1881,6 +1871,15 @@ Decl *TemplateInstantiator::TransformDecl(SourceLocation Loc, Decl *D) { TemplateArgument Arg = TemplateArgs(TTP->getDepth(), TTP->getPosition()); if (TTP->isParameterPack()) { + // We might not have an index for pack expansion when normalizing + // constraint expressions. In that case, resort to instantiation scopes + // for the transformed declarations. + if (SemaRef.ArgumentPackSubstitutionIndex == -1 && + SemaRef.CodeSynthesisContexts.back().Kind == + Sema::CodeSynthesisContext::ConstraintNormalization) { + return SemaRef.FindInstantiatedDecl(Loc, cast(D), + TemplateArgs); + } assert(Arg.getKind() == TemplateArgument::Pack && "Missing argument pack"); Arg = getPackSubstitutedTemplateArgument(getSema(), Arg); diff --git a/clang/test/SemaTemplate/concepts-out-of-line-def.cpp b/clang/test/SemaTemplate/concepts-out-of-line-def.cpp index dd518d283c83c..7cb5cfc10b9a7 100644 --- a/clang/test/SemaTemplate/concepts-out-of-line-def.cpp +++ b/clang/test/SemaTemplate/concepts-out-of-line-def.cpp @@ -703,6 +703,25 @@ C v; } // namespace GH93099 +namespace GH115098 { + +template struct c { + template + requires(sizeof...(Ts) > 0) + friend bool operator==(c, c); +}; + +template struct d { + template + requires(sizeof...(Ts) > 0) + friend bool operator==(d, d); +}; + +template struct c; +template struct d; + +} // namespace GH115098 + namespace GH114685 { template struct ptr {