https://github.com/hekota updated https://github.com/llvm/llvm-project/pull/187132
>From 1a3c736dbe5b44fb2497b29d7f4b853ab5bf280f Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Mon, 2 Mar 2026 18:07:13 -0800 Subject: [PATCH 1/9] [HLSL] Add globals for resources embedded in structs For each resource or resource array member of a struct declared at global scope or inside a cbuffer, create an implicit global variable of the same resource type. The variable name will be derived from the struct instance name and the member name. The new global is associated with the struct declaration using a new attribute HLSLAssociatedResourceDeclAttr. Closes #182988 --- clang/include/clang/AST/HLSLResource.h | 34 ++++ clang/include/clang/Basic/Attr.td | 8 + clang/include/clang/Sema/SemaHLSL.h | 14 +- clang/lib/AST/CMakeLists.txt | 1 + clang/lib/AST/HLSLResource.cpp | 46 +++++ clang/lib/Sema/SemaExprMember.cpp | 6 + clang/lib/Sema/SemaHLSL.cpp | 165 ++++++++++++++++- .../AST/HLSL/resources-in-structs-errors.hlsl | 7 + clang/test/AST/HLSL/resources-in-structs.hlsl | 167 ++++++++++++++++++ 9 files changed, 437 insertions(+), 11 deletions(-) create mode 100644 clang/lib/AST/HLSLResource.cpp create mode 100644 clang/test/AST/HLSL/resources-in-structs-errors.hlsl create mode 100644 clang/test/AST/HLSL/resources-in-structs.hlsl diff --git a/clang/include/clang/AST/HLSLResource.h b/clang/include/clang/AST/HLSLResource.h index 131aebf5f14f2..a37acb3660d00 100644 --- a/clang/include/clang/AST/HLSLResource.h +++ b/clang/include/clang/AST/HLSLResource.h @@ -17,6 +17,7 @@ #include "clang/AST/ASTContext.h" #include "clang/AST/Attr.h" #include "clang/AST/DeclBase.h" +#include "clang/Basic/IdentifierTable.h" #include "clang/Basic/TargetInfo.h" #include "clang/Support/Compiler.h" #include "llvm/Frontend/HLSL/HLSLResource.h" @@ -108,6 +109,39 @@ inline uint32_t getResourceDimensions(llvm::dxil::ResourceDimension Dim) { llvm_unreachable("Unhandled llvm::dxil::ResourceDimension enum."); } +// Helper class for building a name of a global resource variable that +// gets created for a resource embedded in a struct or class. This will +// also be used from CodeGen to build a name that matches the resource +// access with the corresponding declaration. +class EmbeddedResourceNameBuilder { + llvm::SmallString<64> Name; + llvm::SmallVector<unsigned> Offsets; + + inline static constexpr std::string_view BaseClassDelim = "::"; + inline static constexpr std::string_view FieldDelim = "."; + inline static constexpr std::string_view ArrayIndexDelim = FieldDelim; + +public: + EmbeddedResourceNameBuilder(llvm::StringRef BaseName) : Name(BaseName) {} + EmbeddedResourceNameBuilder() : Name("") {} + + void pushName(llvm::StringRef N) { pushName(N, FieldDelim); } + void pushBaseName(llvm::StringRef N); + void pushArrayIndex(uint64_t Index); + + void pop() { + assert(!Offsets.empty() && "no name to pop"); + Name.resize(Offsets.pop_back_val()); + } + + IdentifierInfo *getNameAsIdentifier(ASTContext &AST) const { + return &AST.Idents.get(Name); + } + +private: + void pushName(llvm::StringRef N, llvm::StringRef Delim); +}; + } // namespace hlsl } // namespace clang diff --git a/clang/include/clang/Basic/Attr.td b/clang/include/clang/Basic/Attr.td index a5641e2e008cd..298ac535c8a9b 100644 --- a/clang/include/clang/Basic/Attr.td +++ b/clang/include/clang/Basic/Attr.td @@ -5046,6 +5046,14 @@ def HLSLResourceBinding: InheritableAttr { }]; } +def HLSLAssociatedResourceDecl : InheritableAttr { + let Spellings = []; + let Args = [DeclArgument<Var, "ResDecl">]; + let Subjects = SubjectList<[ExternalGlobalVar], ErrorDiag>; + let LangOpts = [HLSL]; + let Documentation = [InternalOnly]; +} + def HLSLUnparsedSemantic : HLSLAnnotationAttr { let Spellings = []; let Args = [DefaultIntArgument<"Index", 0>, diff --git a/clang/include/clang/Sema/SemaHLSL.h b/clang/include/clang/Sema/SemaHLSL.h index a6a38531ac284..7aa97aa878e7b 100644 --- a/clang/include/clang/Sema/SemaHLSL.h +++ b/clang/include/clang/Sema/SemaHLSL.h @@ -222,6 +222,13 @@ class SemaHLSL : public SemaBase { const IdentifierInfo *CompName, SourceLocation CompLoc); + uint32_t getNextImplicitBindingOrderID() { + return ImplicitBindingNextOrderID++; + } + + bool initGlobalResourceDecl(VarDecl *VD); + bool initGlobalResourceArrayDecl(VarDecl *VD); + private: // HLSL resource type attributes need to be processed all at once. // This is a list to collect them. @@ -315,12 +322,7 @@ class SemaHLSL : public SemaBase { const Attr *A, llvm::Triple::EnvironmentType Stage, IOType CurrentIOType, std::initializer_list<SemanticStageInfo> AllowedStages); - uint32_t getNextImplicitBindingOrderID() { - return ImplicitBindingNextOrderID++; - } - - bool initGlobalResourceDecl(VarDecl *VD); - bool initGlobalResourceArrayDecl(VarDecl *VD); + void handleGlobalStructOrArrayOfWithResources(VarDecl *VD); // Infer a common global binding info for an Expr // diff --git a/clang/lib/AST/CMakeLists.txt b/clang/lib/AST/CMakeLists.txt index f9a5f4f0e7ecd..11d3638e98bc6 100644 --- a/clang/lib/AST/CMakeLists.txt +++ b/clang/lib/AST/CMakeLists.txt @@ -96,6 +96,7 @@ add_clang_library(clangAST ByteCode/State.cpp ByteCode/MemberPointer.cpp ByteCode/InterpShared.cpp + HLSLResource.cpp ItaniumCXXABI.cpp ItaniumMangle.cpp JSONNodeDumper.cpp diff --git a/clang/lib/AST/HLSLResource.cpp b/clang/lib/AST/HLSLResource.cpp new file mode 100644 index 0000000000000..19321625222f3 --- /dev/null +++ b/clang/lib/AST/HLSLResource.cpp @@ -0,0 +1,46 @@ +//===--- HLSLResource.cpp - Routines for HLSL resources and bindings +//-------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// This file provides shared routines to help analyze HLSL resources and +// theirs bindings during Sema and CodeGen. +// +//===----------------------------------------------------------------------===// + +#include "clang/AST/HLSLResource.h" +#include "clang/AST/Decl.h" +#include "clang/AST/DeclCXX.h" +#include "clang/AST/Type.h" + +using namespace clang; + +namespace clang { +namespace hlsl { + +void EmbeddedResourceNameBuilder::pushBaseName(llvm::StringRef N) { + pushName(N, FieldDelim); + Name.append(BaseClassDelim); +} + +void EmbeddedResourceNameBuilder::pushName(llvm::StringRef N, + llvm::StringRef Delim) { + Offsets.push_back(Name.size()); + if (!Name.empty() && !Name.ends_with(BaseClassDelim)) + Name.append(Delim); + Name.append(N); +} + +void EmbeddedResourceNameBuilder::pushArrayIndex(uint64_t Index) { + llvm::raw_svector_ostream OS(Name); + Offsets.push_back(Name.size()); + OS << ArrayIndexDelim; + OS << Index; +} + +} // namespace hlsl +} // namespace clang diff --git a/clang/lib/Sema/SemaExprMember.cpp b/clang/lib/Sema/SemaExprMember.cpp index e2f26ef5aa2b2..ff7c844f66123 100644 --- a/clang/lib/Sema/SemaExprMember.cpp +++ b/clang/lib/Sema/SemaExprMember.cpp @@ -1812,6 +1812,12 @@ Sema::BuildFieldReferenceExpr(Expr *BaseExpr, bool IsArrow, // except that 'mutable' members don't pick up 'const'. if (Field->isMutable()) BaseQuals.removeConst(); + // HLSL resource types do not pick up address space qualifiers from the + // base. + if (getLangOpts().HLSL && (MemberType->isHLSLResourceRecord() || + MemberType->isHLSLResourceRecordArray())) + BaseQuals.removeAddressSpace(); + Qualifiers MemberQuals = Context.getCanonicalType(MemberType).getQualifiers(); diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index 804ea70aaddce..df38154faeaa0 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -342,21 +342,29 @@ static bool isZeroSizedArray(const ConstantArrayType *CAT) { return CAT != nullptr; } -static bool isResourceRecordTypeOrArrayOf(VarDecl *VD) { - const Type *Ty = VD->getType().getTypePtr(); +static bool isResourceRecordTypeOrArrayOf(QualType Ty) { return Ty->isHLSLResourceRecord() || Ty->isHLSLResourceRecordArray(); } +static bool isResourceRecordTypeOrArrayOf(VarDecl *VD) { + return isResourceRecordTypeOrArrayOf(VD->getType()); +} + static const HLSLAttributedResourceType * -getResourceArrayHandleType(VarDecl *VD) { - assert(VD->getType()->isHLSLResourceRecordArray() && +getResourceArrayHandleType(QualType QT) { + assert(QT->isHLSLResourceRecordArray() && "expected array of resource records"); - const Type *Ty = VD->getType()->getUnqualifiedDesugaredType(); + const Type *Ty = QT->getUnqualifiedDesugaredType(); while (const ArrayType *AT = dyn_cast<ArrayType>(Ty)) Ty = AT->getArrayElementTypeNoTypeQual()->getUnqualifiedDesugaredType(); return HLSLAttributedResourceType::findHandleTypeOnResource(Ty); } +static const HLSLAttributedResourceType * +getResourceArrayHandleType(VarDecl *VD) { + return getResourceArrayHandleType(VD->getType()); +} + // Returns true if the type is a leaf element type that is not valid to be // included in HLSL Buffer, such as a resource class, empty struct, zero-sized // array, or a builtin intangible type. Returns false it is a valid leaf element @@ -4438,6 +4446,147 @@ void SemaHLSL::deduceAddressSpace(VarDecl *Decl) { Decl->setType(Type); } +namespace { + +static void handleResourceFieldsInStruct( + Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, + EmbeddedResourceNameBuilder &NameBuilder); + +static void handleStructWithResources(Sema &S, VarDecl *ParentVD, + const CXXRecordDecl *RD, + EmbeddedResourceNameBuilder &NameBuilder) { + + // scan the base classes + assert(RD->getNumBases() <= 1 && "HLSL doesn't support multiple inheritance"); + const auto *BasesIt = RD->bases_begin(); + if (BasesIt != RD->bases_end()) { + QualType QT = BasesIt->getType(); + if (QT->isHLSLIntangibleType()) { + CXXRecordDecl *BaseRD = QT->getAsCXXRecordDecl(); + NameBuilder.pushBaseName(BaseRD->getName()); + handleStructWithResources(S, ParentVD, BaseRD, NameBuilder); + NameBuilder.pop(); + } + } + // process this class fields + handleResourceFieldsInStruct(S, ParentVD, RD, NameBuilder); +} + +static void +handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, + const ConstantArrayType *CAT, + EmbeddedResourceNameBuilder &NameBuilder) { + + QualType ElementTy = CAT->getElementType().getCanonicalType(); + assert(ElementTy->isHLSLIntangibleType() && "Expected HLSL intangible type"); + + const ConstantArrayType *SubCAT = dyn_cast<ConstantArrayType>(ElementTy); + const CXXRecordDecl *ElementRD = ElementTy->getAsCXXRecordDecl(); + assert((SubCAT || ElementRD) && + "Expected struct type or an constant array of structs"); + + for (unsigned I = 0, E = CAT->getSize().getZExtValue(); I < E; ++I) { + NameBuilder.pushArrayIndex(I); + if (ElementRD) + handleStructWithResources(S, ParentVD, ElementRD, NameBuilder); + else + handleArrayOfStructWithResources(S, ParentVD, SubCAT, NameBuilder); + NameBuilder.pop(); + } +} + +static void createGlobalResourceDeclForStruct( + Sema &S, VarDecl *ParentVD, SourceLocation Loc, IdentifierInfo *Id, + QualType ResTy) { + assert(isResourceRecordTypeOrArrayOf(ResTy) && + "expected resource type or array of resources"); + + DeclContext *DC = ParentVD->getNonTransparentDeclContext(); + assert(DC->isTranslationUnit() && "expected translation unit decl context"); + + ASTContext &AST = S.getASTContext(); + VarDecl *ResDecl = + VarDecl::Create(AST, DC, Loc, Loc, Id, ResTy, nullptr, SC_None); + + unsigned Range = 1; + const HLSLAttributedResourceType *ResHandleTy = nullptr; + if (const auto *AT = dyn_cast<ArrayType>(ResTy.getTypePtr())) { + const auto *CAT = dyn_cast<ConstantArrayType>(AT); + Range = CAT ? CAT->getSize().getZExtValue() : -1; + ResHandleTy = getResourceArrayHandleType(ResTy); + } else { + ResHandleTy = HLSLAttributedResourceType::findHandleTypeOnResource( + ResTy.getTypePtr()); + } + // FIXME: Explicit bindings will be handled in a follow-up change. For now + // just add an implicit binding attribute. + auto *Attr = + HLSLResourceBindingAttr::CreateImplicit(S.getASTContext(), "", "0", {}); + Attr->setBinding(getRegisterType(ResHandleTy), std::nullopt, 0); + Attr->setImplicitBindingOrderID(S.HLSL().getNextImplicitBindingOrderID()); + ResDecl->addAttr(Attr); + ResDecl->setImplicit(); + + if (Range == 1) + S.HLSL().initGlobalResourceDecl(ResDecl); + else + S.HLSL().initGlobalResourceArrayDecl(ResDecl); + + ParentVD->addAttr( + HLSLAssociatedResourceDeclAttr::CreateImplicit(AST, ResDecl)); + DC->addDecl(ResDecl); + + DeclGroupRef DG(ResDecl); + S.Consumer.HandleTopLevelDecl(DG); +} + +static void +handleResourceFieldsInStruct(Sema &S, VarDecl *ParentVD, + const CXXRecordDecl *RD, + EmbeddedResourceNameBuilder &NameBuilder) { + + for (const FieldDecl *FD : RD->fields()) { + QualType FDTy = FD->getType().getCanonicalType(); + if (!FDTy->isHLSLIntangibleType()) + continue; + + NameBuilder.pushName(FD->getName()); + + if (isResourceRecordTypeOrArrayOf(FDTy)) { + IdentifierInfo *II = NameBuilder.getNameAsIdentifier(S.getASTContext()); + createGlobalResourceDeclForStruct(S, ParentVD, FD->getLocation(), II, + FDTy); + } else if (const auto *RD = FDTy->getAsCXXRecordDecl()) { + handleStructWithResources(S, ParentVD, RD, NameBuilder); + + } else if (const auto *ArrayTy = dyn_cast<ConstantArrayType>(FDTy)) { + assert(!FDTy->isHLSLResourceRecordArray() && + "resource arrays should have been already handled"); + handleArrayOfStructWithResources(S, ParentVD, ArrayTy, NameBuilder); + } + NameBuilder.pop(); + } +} + +} // namespace + +void SemaHLSL::handleGlobalStructOrArrayOfWithResources(VarDecl *VD) { + EmbeddedResourceNameBuilder NameBuilder(VD->getName()); + + const Type *VDTy = VD->getType().getTypePtr(); + const CXXRecordDecl *RD = VDTy->getAsCXXRecordDecl(); + if (RD) { + handleStructWithResources(SemaRef, VD, RD, NameBuilder); + return; + } + + const auto *CAT = dyn_cast<ConstantArrayType>(VDTy); + if (CAT) { + handleArrayOfStructWithResources(SemaRef, VD, CAT, NameBuilder); + return; + } +} + void SemaHLSL::ActOnVariableDeclarator(VarDecl *VD) { if (VD->hasGlobalStorage()) { // make sure the declaration has a complete type @@ -4510,6 +4659,12 @@ void SemaHLSL::ActOnVariableDeclarator(VarDecl *VD) { } } } + + // Process resources in user-defined structs, or arrays of such structs. + const Type *VDTy = VD->getType().getTypePtr(); + if (VD->getStorageClass() != SC_Static && VDTy->isHLSLIntangibleType() && + !isResourceRecordTypeOrArrayOf(VD)) + handleGlobalStructOrArrayOfWithResources(VD); } deduceAddressSpace(VD); diff --git a/clang/test/AST/HLSL/resources-in-structs-errors.hlsl b/clang/test/AST/HLSL/resources-in-structs-errors.hlsl new file mode 100644 index 0000000000000..56baef7957fb1 --- /dev/null +++ b/clang/test/AST/HLSL/resources-in-structs-errors.hlsl @@ -0,0 +1,7 @@ +// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -verify %s + +struct A { + RWBuffer<float> Buf; +}; + +A incompleteArray[]; // expected-error {{definition of variable with array type needs an explicit size or an initializer}} diff --git a/clang/test/AST/HLSL/resources-in-structs.hlsl b/clang/test/AST/HLSL/resources-in-structs.hlsl new file mode 100644 index 0000000000000..f6e8b26787adc --- /dev/null +++ b/clang/test/AST/HLSL/resources-in-structs.hlsl @@ -0,0 +1,167 @@ +// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -ast-dump %s | FileCheck %s + +// Single resource field in struct + +// CHECK: CXXRecordDecl {{.*}} struct A +// CHECK: FieldDecl {{.*}} Buf 'RWBuffer<float>':'hlsl::RWBuffer<float>' +struct A { + RWBuffer<float> Buf; +}; + +// CHECK: VarDecl {{.*}} implicit a1.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} a1 'hlsl_constant A' +// CHECK: HLSLResourceBindingAttr {{.*}} "u0" "space0" +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'a1.Buf' 'hlsl::RWBuffer<float>' +A a1 : register(u0); + +// Resource array in struct + +// CHECK: CXXRecordDecl {{.*}} struct B +// CHECK: FieldDecl {{.*}} Bufs 'RWBuffer<float>[10]' +struct B { + RWBuffer<float> Bufs[10]; +}; + +// CHECK: VarDecl {{.*}} implicit b1.Bufs 'hlsl::RWBuffer<float>[10]' +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} b1 'hlsl_constant B' +// CHECK: HLSLResourceBindingAttr {{.*}} "u2" "space0" +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'b1.Bufs' 'hlsl::RWBuffer<float>[10]' +B b1 : register(u2); + +// Inheritance + +// CHECK: CXXRecordDecl {{.*}} struct C +// CHECK: FieldDecl {{.*}} Buf2 'RWBuffer<float>':'hlsl::RWBuffer<float>' +struct C : A { + RWBuffer<float> Buf2; +}; + +// CHECK: VarDecl {{.*}} implicit c1.A::Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit c1.Buf2 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} c1 'hlsl_constant C' +// CHECK: HLSLResourceBindingAttr {{.*}} "u3" "space0" +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'c1.A::Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'c1.Buf2' 'hlsl::RWBuffer<float>' +C c1 : register(u3); + +// Inheritance with same named field +// CHECK: CXXRecordDecl {{.*}} struct D +// CHECK: FieldDecl {{.*}} A 'A' +struct D : A { + A A; +}; + +// CHECK: VarDecl {{.*}} implicit d1.A::Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit d1.A.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} d1 'hlsl_constant D' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'd1.A::Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'd1.A.Buf' 'hlsl::RWBuffer<float>' +D d1; + +// Inheritance and Multiple Resources Kinds + +// CHECK: CXXRecordDecl {{.*}} class E +// CHECK: FieldDecl {{.*}} SrvBuf 'StructuredBuffer<int>':'hlsl::StructuredBuffer<int>' +class E { + StructuredBuffer<int> SrvBuf; +}; + +// CHECK: CXXRecordDecl {{.*}} class F +// CHECK: FieldDecl {{.*}} a 'A' +// CHECK: FieldDecl {{.*}} SrvBuf 'StructuredBuffer<float>':'hlsl::StructuredBuffer<float>' +// CHECK: FieldDecl {{.*}} Samp 'SamplerState' +class F : E { + A a; + StructuredBuffer<float> SrvBuf; + SamplerState Samp; +}; + +// CHECK: VarDecl {{.*}} implicit f.E::SrvBuf 'hlsl::StructuredBuffer<int>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit f.a.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit f.SrvBuf 'hlsl::StructuredBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit f.Samp 'hlsl::SamplerState' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} f 'hlsl_constant F' +// CHECK: HLSLResourceBindingAttr {{.*}} "t0" "space0" +// CHECK: HLSLResourceBindingAttr {{.*}} "u20" "space0" +// CHECK: HLSLResourceBindingAttr {{.*}} "s3" "space0" +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.E::SrvBuf' 'hlsl::StructuredBuffer<int>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.a.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.SrvBuf' 'hlsl::StructuredBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.Samp' 'hlsl::SamplerState' +F f : register(t0) : register(u20) : register(s3); + +// Array of structs with resources + +// CHECK: VarDecl {{.*}} implicit arrayOfA.0.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit arrayOfA.1.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +A arrayOfA[2] : register(u0, space1); + +// CHECK: CXXRecordDecl {{.*}} struct G +// CHECK: FieldDecl {{.*}} multiArray 'A[2][2]' +struct G { + A multiArray[2][2]; +}; + +// CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.0.0.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.0.1.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.1.0.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.1.1.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.0.0.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.0.1.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.1.0.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.1.1.Buf 'hlsl::RWBuffer<float>' callinit +// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" + +// CHECK: VarDecl {{.*}} gArray 'hlsl_constant G[2]' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.0.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.0.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.1.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.1.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.0.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.0.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.1.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.1.1.Buf' 'hlsl::RWBuffer<float>' +G gArray[2] : register(u10, space2); + +// Static struct with resources + +// CHECK-NOT: VarDecl {{.*}} a2.Buf +// CHECK: VarDecl {{.*}} a2 'hlsl_private A' static cinit +static A a2 = { a1 }; >From 908782fafbd03905482d239c285047d1fe2df6a7 Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Mon, 2 Mar 2026 18:50:41 -0800 Subject: [PATCH 2/9] Reorder and format --- clang/lib/Sema/SemaHLSL.cpp | 100 +++++++++++++++++------------------- 1 file changed, 47 insertions(+), 53 deletions(-) diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index df38154faeaa0..e397d32643857 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -4448,56 +4448,10 @@ void SemaHLSL::deduceAddressSpace(VarDecl *Decl) { namespace { -static void handleResourceFieldsInStruct( - Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, - EmbeddedResourceNameBuilder &NameBuilder); - -static void handleStructWithResources(Sema &S, VarDecl *ParentVD, - const CXXRecordDecl *RD, - EmbeddedResourceNameBuilder &NameBuilder) { - - // scan the base classes - assert(RD->getNumBases() <= 1 && "HLSL doesn't support multiple inheritance"); - const auto *BasesIt = RD->bases_begin(); - if (BasesIt != RD->bases_end()) { - QualType QT = BasesIt->getType(); - if (QT->isHLSLIntangibleType()) { - CXXRecordDecl *BaseRD = QT->getAsCXXRecordDecl(); - NameBuilder.pushBaseName(BaseRD->getName()); - handleStructWithResources(S, ParentVD, BaseRD, NameBuilder); - NameBuilder.pop(); - } - } - // process this class fields - handleResourceFieldsInStruct(S, ParentVD, RD, NameBuilder); -} - -static void -handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, - const ConstantArrayType *CAT, - EmbeddedResourceNameBuilder &NameBuilder) { - - QualType ElementTy = CAT->getElementType().getCanonicalType(); - assert(ElementTy->isHLSLIntangibleType() && "Expected HLSL intangible type"); - - const ConstantArrayType *SubCAT = dyn_cast<ConstantArrayType>(ElementTy); - const CXXRecordDecl *ElementRD = ElementTy->getAsCXXRecordDecl(); - assert((SubCAT || ElementRD) && - "Expected struct type or an constant array of structs"); - - for (unsigned I = 0, E = CAT->getSize().getZExtValue(); I < E; ++I) { - NameBuilder.pushArrayIndex(I); - if (ElementRD) - handleStructWithResources(S, ParentVD, ElementRD, NameBuilder); - else - handleArrayOfStructWithResources(S, ParentVD, SubCAT, NameBuilder); - NameBuilder.pop(); - } -} - -static void createGlobalResourceDeclForStruct( - Sema &S, VarDecl *ParentVD, SourceLocation Loc, IdentifierInfo *Id, - QualType ResTy) { +static void createGlobalResourceDeclForStruct(Sema &S, VarDecl *ParentVD, + SourceLocation Loc, + IdentifierInfo *Id, + QualType ResTy) { assert(isResourceRecordTypeOrArrayOf(ResTy) && "expected resource type or array of resources"); @@ -4541,10 +4495,27 @@ static void createGlobalResourceDeclForStruct( } static void -handleResourceFieldsInStruct(Sema &S, VarDecl *ParentVD, - const CXXRecordDecl *RD, - EmbeddedResourceNameBuilder &NameBuilder) { +handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, + const ConstantArrayType *CAT, + EmbeddedResourceNameBuilder &NameBuilder); +static void +handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, + EmbeddedResourceNameBuilder &NameBuilder) { + + // scan the base classes + assert(RD->getNumBases() <= 1 && "HLSL doesn't support multiple inheritance"); + const auto *BasesIt = RD->bases_begin(); + if (BasesIt != RD->bases_end()) { + QualType QT = BasesIt->getType(); + if (QT->isHLSLIntangibleType()) { + CXXRecordDecl *BaseRD = QT->getAsCXXRecordDecl(); + NameBuilder.pushBaseName(BaseRD->getName()); + handleStructWithResources(S, ParentVD, BaseRD, NameBuilder); + NameBuilder.pop(); + } + } + // process this class fields for (const FieldDecl *FD : RD->fields()) { QualType FDTy = FD->getType().getCanonicalType(); if (!FDTy->isHLSLIntangibleType()) @@ -4568,6 +4539,29 @@ handleResourceFieldsInStruct(Sema &S, VarDecl *ParentVD, } } +static void +handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, + const ConstantArrayType *CAT, + EmbeddedResourceNameBuilder &NameBuilder) { + + QualType ElementTy = CAT->getElementType().getCanonicalType(); + assert(ElementTy->isHLSLIntangibleType() && "Expected HLSL intangible type"); + + const ConstantArrayType *SubCAT = dyn_cast<ConstantArrayType>(ElementTy); + const CXXRecordDecl *ElementRD = ElementTy->getAsCXXRecordDecl(); + assert((SubCAT || ElementRD) && + "Expected struct type or an constant array of structs"); + + for (unsigned I = 0, E = CAT->getSize().getZExtValue(); I < E; ++I) { + NameBuilder.pushArrayIndex(I); + if (ElementRD) + handleStructWithResources(S, ParentVD, ElementRD, NameBuilder); + else + handleArrayOfStructWithResources(S, ParentVD, SubCAT, NameBuilder); + NameBuilder.pop(); + } +} + } // namespace void SemaHLSL::handleGlobalStructOrArrayOfWithResources(VarDecl *VD) { >From 64141ab75a2554da799392fc4eada65d32b8c263 Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Tue, 3 Mar 2026 13:22:03 -0800 Subject: [PATCH 3/9] Add comments, update asserts --- clang/lib/Sema/SemaHLSL.cpp | 24 ++++++++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index e397d32643857..fa24ac31ad646 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -4448,6 +4448,9 @@ void SemaHLSL::deduceAddressSpace(VarDecl *Decl) { namespace { +// Creates a global variable declaration for a resource field embedded in a +// struct, assigns it a binding, initializes it, and associates it with the +// struct declaration via an HLSLAssociatedResourceDeclAttr. static void createGlobalResourceDeclForStruct(Sema &S, VarDecl *ParentVD, SourceLocation Loc, IdentifierInfo *Id, @@ -4499,11 +4502,14 @@ handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, const ConstantArrayType *CAT, EmbeddedResourceNameBuilder &NameBuilder); +// Scans base and all fields of a struct/class type to find all embedded +// resources or resource arrays,. Creates a global variable for each resource +// found. static void handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, EmbeddedResourceNameBuilder &NameBuilder) { - // scan the base classes + // Scan the base classes. assert(RD->getNumBases() <= 1 && "HLSL doesn't support multiple inheritance"); const auto *BasesIt = RD->bases_begin(); if (BasesIt != RD->bases_end()) { @@ -4515,7 +4521,7 @@ handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, NameBuilder.pop(); } } - // process this class fields + // Process this class fields. for (const FieldDecl *FD : RD->fields()) { QualType FDTy = FD->getType().getCanonicalType(); if (!FDTy->isHLSLIntangibleType()) @@ -4539,6 +4545,7 @@ handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, } } +// Processes array of structs with resources. static void handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, const ConstantArrayType *CAT, @@ -4549,8 +4556,9 @@ handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, const ConstantArrayType *SubCAT = dyn_cast<ConstantArrayType>(ElementTy); const CXXRecordDecl *ElementRD = ElementTy->getAsCXXRecordDecl(); - assert((SubCAT || ElementRD) && - "Expected struct type or an constant array of structs"); + + if (!SubCAT && !ElementRD) + return; for (unsigned I = 0, E = CAT->getSize().getZExtValue(); I < E; ++I) { NameBuilder.pushArrayIndex(I); @@ -4564,10 +4572,18 @@ handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, } // namespace +// Scans all fields of a user-defined struct (or array of structs) +// to find all embedded resources or resource arrays. For each resource +// a global variable of the resource type is created and associated +// with the parent declaration (VD) through a HLSLAssociatedResourceDeclAttr +// attribute. void SemaHLSL::handleGlobalStructOrArrayOfWithResources(VarDecl *VD) { EmbeddedResourceNameBuilder NameBuilder(VD->getName()); const Type *VDTy = VD->getType().getTypePtr(); + assert(VDTy->isHLSLIntangibleType() && !isResourceRecordTypeOrArrayOf(VD) && + "Expected non-resource struct or array type"); + const CXXRecordDecl *RD = VDTy->getAsCXXRecordDecl(); if (RD) { handleStructWithResources(SemaRef, VD, RD, NameBuilder); >From 70ca5d5d63db769943fb6fea7de41f50a2600223 Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Wed, 4 Mar 2026 18:31:11 -0800 Subject: [PATCH 4/9] [HLSL] Add binding attributes to resources in structs Add binding attributes to global variables that were created for resources embedded in structs. The binding values are based on `register` annotations and `[[vk::binding]]` attribute on the struct instance. Depends on #184281 Fixes #182992 --- clang/lib/Sema/SemaHLSL.cpp | 170 +++++++++++++++--- clang/test/AST/HLSL/resources-in-structs.hlsl | 125 ++++++++----- 2 files changed, 224 insertions(+), 71 deletions(-) diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index fa24ac31ad646..805b97dc858d9 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -111,6 +111,24 @@ static bool convertToRegisterType(StringRef Slot, RegisterType *RT) { } } +static char getRegisterTypeChar(RegisterType RT) { + switch (RT) { + case RegisterType::SRV: + return 't'; + case RegisterType::UAV: + return 'u'; + case RegisterType::CBuffer: + return 'b'; + case RegisterType::Sampler: + return 's'; + case RegisterType::C: + return 'c'; + case RegisterType::I: + return 'i'; + } + llvm_unreachable("unexpected RegisterType value"); +} + static ResourceClass getResourceClass(RegisterType RT) { switch (RT) { case RegisterType::SRV: @@ -163,6 +181,15 @@ static Builtin::ID getSpecConstBuiltinId(const Type *Type) { } } +static StringRef createRegisterString(ASTContext &AST, RegisterType RegType, + unsigned N) { + llvm::SmallString<16> Buffer; + llvm::raw_svector_ostream OS(Buffer); + OS << getRegisterTypeChar(RegType); + OS << N; + return AST.backupStr(OS.str()); +} + DeclBindingInfo *ResourceBindings::addDeclBindingInfo(const VarDecl *VD, ResourceClass ResClass) { assert(getDeclBindingInfo(VD, ResClass) == nullptr && @@ -4448,13 +4475,99 @@ void SemaHLSL::deduceAddressSpace(VarDecl *Decl) { namespace { +// Helper class for assigning bindings to resources declared within a struct. +// It keeps track of all binding attributes declared on a struct instance, and +// the offsets for each register type that have been assigned so far. +// Handles both explicit and implicit bindings. +class StructBindingContext { + // Bindings and offsets per register type. We only need to support four + // register types - SRV (u), UAV (t), CBuffer (c), and Sampler (s). + HLSLResourceBindingAttr *RegBindingsAttrs[4]; + unsigned RegBindingOffset[4]; + + // Vulkan binding attribute does not vary by register type. + HLSLVkBindingAttr *VkBindingAttr; + unsigned VkBindingOffset; + +public: + // Constructor: gather all binding attributes on a struct instance and + // initialize offsets. + StructBindingContext(VarDecl *VD) { + for (unsigned i = 0; i < 4; ++i) { + RegBindingsAttrs[i] = nullptr; + RegBindingOffset[i] = 0; + } + VkBindingAttr = nullptr; + VkBindingOffset = 0; + + ASTContext &AST = VD->getASTContext(); + bool IsSpirv = AST.getTargetInfo().getTriple().isSPIRV(); + + for (Attr *A : VD->attrs()) { + if (auto *RBA = dyn_cast<HLSLResourceBindingAttr>(A)) { + RegisterType RegType = RBA->getRegisterType(); + unsigned RegTypeIdx = static_cast<unsigned>(RegType); + // Ignore unsupported register annotations, such as 'c' or 'i'. + if (RegTypeIdx < 4) + RegBindingsAttrs[RegTypeIdx] = RBA; + continue; + } + // Gather the Vulkan binding attributes only if the target is SPIR-V. + if (IsSpirv) { + if (auto *VBA = dyn_cast<HLSLVkBindingAttr>(A)) + VkBindingAttr = VBA; + } + } + } + + // Creates a binding attribute for a resource based on the gathered attributes + // and the required register type and range. + Attr *createBindingAttr(SemaHLSL &S, ASTContext &AST, RegisterType RegType, + unsigned Range) { + assert(static_cast<unsigned>(RegType) < 4 && "unexpected register type"); + + if (VkBindingAttr) { + unsigned Offset = VkBindingOffset; + VkBindingOffset += Range; + return HLSLVkBindingAttr::CreateImplicit( + AST, VkBindingAttr->getBinding() + Offset, VkBindingAttr->getSet(), + VkBindingAttr->getRange()); + } + + HLSLResourceBindingAttr *RBA = + RegBindingsAttrs[static_cast<unsigned>(RegType)]; + HLSLResourceBindingAttr *NewAttr = nullptr; + + if (RBA && RBA->hasRegisterSlot()) { + // Explicit binding - create a new attribute with offseted slot number + // based on the required register type. + unsigned Offset = RegBindingOffset[static_cast<unsigned>(RegType)]; + RegBindingOffset[static_cast<unsigned>(RegType)] += Range; + + unsigned NewSlotNumber = RBA->getSlotNumber() + Offset; + StringRef NewSlotNumberStr = + createRegisterString(AST, RBA->getRegisterType(), NewSlotNumber); + NewAttr = HLSLResourceBindingAttr::CreateImplicit( + AST, NewSlotNumberStr, RBA->getSpace(), RBA->getRange()); + NewAttr->setBinding(RegType, NewSlotNumber, RBA->getSpaceNumber()); + } else { + // No binding attribute or space-only binding - create a binding + // attribute for implicit binding. + NewAttr = HLSLResourceBindingAttr::CreateImplicit(AST, "", "0", {}); + NewAttr->setBinding(RegType, std::nullopt, + RBA ? RBA->getSpaceNumber() : 0); + NewAttr->setImplicitBindingOrderID(S.getNextImplicitBindingOrderID()); + } + return NewAttr; + } +}; + // Creates a global variable declaration for a resource field embedded in a // struct, assigns it a binding, initializes it, and associates it with the // struct declaration via an HLSLAssociatedResourceDeclAttr. -static void createGlobalResourceDeclForStruct(Sema &S, VarDecl *ParentVD, - SourceLocation Loc, - IdentifierInfo *Id, - QualType ResTy) { +static void createGlobalResourceDeclForStruct( + Sema &S, VarDecl *ParentVD, SourceLocation Loc, IdentifierInfo *Id, + QualType ResTy, StructBindingContext &BindingCtx) { assert(isResourceRecordTypeOrArrayOf(ResTy) && "expected resource type or array of resources"); @@ -4475,13 +4588,11 @@ static void createGlobalResourceDeclForStruct(Sema &S, VarDecl *ParentVD, ResHandleTy = HLSLAttributedResourceType::findHandleTypeOnResource( ResTy.getTypePtr()); } - // FIXME: Explicit bindings will be handled in a follow-up change. For now - // just add an implicit binding attribute. - auto *Attr = - HLSLResourceBindingAttr::CreateImplicit(S.getASTContext(), "", "0", {}); - Attr->setBinding(getRegisterType(ResHandleTy), std::nullopt, 0); - Attr->setImplicitBindingOrderID(S.HLSL().getNextImplicitBindingOrderID()); - ResDecl->addAttr(Attr); + // Add a binding attribute to the global resource declaration. + Attr *BindingAttr = BindingCtx.createBindingAttr( + S.HLSL(), AST, getRegisterType(ResHandleTy), Range); + ResDecl->addAttr(BindingAttr); + ResDecl->addAttr(InternalLinkageAttr::CreateImplicit(AST)); ResDecl->setImplicit(); if (Range == 1) @@ -4497,17 +4608,17 @@ static void createGlobalResourceDeclForStruct(Sema &S, VarDecl *ParentVD, S.Consumer.HandleTopLevelDecl(DG); } -static void -handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, - const ConstantArrayType *CAT, - EmbeddedResourceNameBuilder &NameBuilder); +static void handleArrayOfStructWithResources( + Sema &S, VarDecl *ParentVD, const ConstantArrayType *CAT, + EmbeddedResourceNameBuilder &NameBuilder, StructBindingContext &BindingCtx); // Scans base and all fields of a struct/class type to find all embedded // resources or resource arrays,. Creates a global variable for each resource // found. -static void -handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, - EmbeddedResourceNameBuilder &NameBuilder) { +static void handleStructWithResources(Sema &S, VarDecl *ParentVD, + const CXXRecordDecl *RD, + EmbeddedResourceNameBuilder &NameBuilder, + StructBindingContext &BindingCtx) { // Scan the base classes. assert(RD->getNumBases() <= 1 && "HLSL doesn't support multiple inheritance"); @@ -4517,7 +4628,7 @@ handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, if (QT->isHLSLIntangibleType()) { CXXRecordDecl *BaseRD = QT->getAsCXXRecordDecl(); NameBuilder.pushBaseName(BaseRD->getName()); - handleStructWithResources(S, ParentVD, BaseRD, NameBuilder); + handleStructWithResources(S, ParentVD, BaseRD, NameBuilder, BindingCtx); NameBuilder.pop(); } } @@ -4532,14 +4643,15 @@ handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, if (isResourceRecordTypeOrArrayOf(FDTy)) { IdentifierInfo *II = NameBuilder.getNameAsIdentifier(S.getASTContext()); createGlobalResourceDeclForStruct(S, ParentVD, FD->getLocation(), II, - FDTy); + FDTy, BindingCtx); } else if (const auto *RD = FDTy->getAsCXXRecordDecl()) { - handleStructWithResources(S, ParentVD, RD, NameBuilder); + handleStructWithResources(S, ParentVD, RD, NameBuilder, BindingCtx); } else if (const auto *ArrayTy = dyn_cast<ConstantArrayType>(FDTy)) { assert(!FDTy->isHLSLResourceRecordArray() && "resource arrays should have been already handled"); - handleArrayOfStructWithResources(S, ParentVD, ArrayTy, NameBuilder); + handleArrayOfStructWithResources(S, ParentVD, ArrayTy, NameBuilder, + BindingCtx); } NameBuilder.pop(); } @@ -4549,7 +4661,8 @@ handleStructWithResources(Sema &S, VarDecl *ParentVD, const CXXRecordDecl *RD, static void handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, const ConstantArrayType *CAT, - EmbeddedResourceNameBuilder &NameBuilder) { + EmbeddedResourceNameBuilder &NameBuilder, + StructBindingContext &BindingCtx) { QualType ElementTy = CAT->getElementType().getCanonicalType(); assert(ElementTy->isHLSLIntangibleType() && "Expected HLSL intangible type"); @@ -4563,9 +4676,11 @@ handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, for (unsigned I = 0, E = CAT->getSize().getZExtValue(); I < E; ++I) { NameBuilder.pushArrayIndex(I); if (ElementRD) - handleStructWithResources(S, ParentVD, ElementRD, NameBuilder); + handleStructWithResources(S, ParentVD, ElementRD, NameBuilder, + BindingCtx); else - handleArrayOfStructWithResources(S, ParentVD, SubCAT, NameBuilder); + handleArrayOfStructWithResources(S, ParentVD, SubCAT, NameBuilder, + BindingCtx); NameBuilder.pop(); } } @@ -4578,6 +4693,7 @@ handleArrayOfStructWithResources(Sema &S, VarDecl *ParentVD, // with the parent declaration (VD) through a HLSLAssociatedResourceDeclAttr // attribute. void SemaHLSL::handleGlobalStructOrArrayOfWithResources(VarDecl *VD) { + StructBindingContext BindingCtx(VD); EmbeddedResourceNameBuilder NameBuilder(VD->getName()); const Type *VDTy = VD->getType().getTypePtr(); @@ -4586,13 +4702,13 @@ void SemaHLSL::handleGlobalStructOrArrayOfWithResources(VarDecl *VD) { const CXXRecordDecl *RD = VDTy->getAsCXXRecordDecl(); if (RD) { - handleStructWithResources(SemaRef, VD, RD, NameBuilder); + handleStructWithResources(SemaRef, VD, RD, NameBuilder, BindingCtx); return; } const auto *CAT = dyn_cast<ConstantArrayType>(VDTy); if (CAT) { - handleArrayOfStructWithResources(SemaRef, VD, CAT, NameBuilder); + handleArrayOfStructWithResources(SemaRef, VD, CAT, NameBuilder, BindingCtx); return; } } diff --git a/clang/test/AST/HLSL/resources-in-structs.hlsl b/clang/test/AST/HLSL/resources-in-structs.hlsl index f6e8b26787adc..6f16633ce8620 100644 --- a/clang/test/AST/HLSL/resources-in-structs.hlsl +++ b/clang/test/AST/HLSL/resources-in-structs.hlsl @@ -1,4 +1,5 @@ -// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -ast-dump %s | FileCheck %s +// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -ast-dump %s | FileCheck %s -check-prefixes=CHECK,DXIL +// RUN: %clang_cc1 -triple spirv-unknown-vulkan-compute -ast-dump %s | FileCheck %s -check-prefixes=CHECK,SPIRV // Single resource field in struct @@ -9,11 +10,14 @@ struct A { }; // CHECK: VarDecl {{.*}} implicit a1.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 0 0 +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u0" "space0" // CHECK: VarDecl {{.*}} a1 'hlsl_constant A' -// CHECK: HLSLResourceBindingAttr {{.*}} "u0" "space0" +// CHECK-NEXT: HLSLVkBindingAttr {{.*}} 0 0 +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "u0" "space0" // CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'a1.Buf' 'hlsl::RWBuffer<float>' +[[vk::binding(0)]] A a1 : register(u0); // Resource array in struct @@ -24,13 +28,17 @@ struct B { RWBuffer<float> Bufs[10]; }; -// CHECK: VarDecl {{.*}} implicit b1.Bufs 'hlsl::RWBuffer<float>[10]' -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// Check when the struct has only [[vk::binding]] binding attribute. + +// CHECK: VarDecl {{.*}} b1.Bufs 'hlsl::RWBuffer<float>[10]' +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 2 0 // CHECK: VarDecl {{.*}} b1 'hlsl_constant B' -// CHECK: HLSLResourceBindingAttr {{.*}} "u2" "space0" +// CHECK-NEXT: HLSLVkBindingAttr {{.*}} 2 0 // CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'b1.Bufs' 'hlsl::RWBuffer<float>[10]' -B b1 : register(u2); +[[vk::binding(2)]] +B b1; // Inheritance @@ -41,15 +49,19 @@ struct C : A { }; // CHECK: VarDecl {{.*}} implicit c1.A::Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u3" "space0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 3 0 // CHECK: VarDecl {{.*}} implicit c1.Buf2 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u4" "space0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 4 0 // CHECK: VarDecl {{.*}} c1 'hlsl_constant C' -// CHECK: HLSLResourceBindingAttr {{.*}} "u3" "space0" -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'c1.A::Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'c1.Buf2' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLVkBindingAttr {{.*}} 3 0 +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "u3" "space0" +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'c1.A::Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'c1.Buf2' 'hlsl::RWBuffer<float>' +[[vk::binding(3)]] C c1 : register(u3); // Inheritance with same named field @@ -66,8 +78,8 @@ struct D : A { // CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" // CHECK: VarDecl {{.*}} d1 'hlsl_constant D' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'd1.A::Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'd1.A.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'd1.A::Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'd1.A.Buf' 'hlsl::RWBuffer<float>' D d1; // Inheritance and Multiple Resources Kinds @@ -89,34 +101,49 @@ class F : E { }; // CHECK: VarDecl {{.*}} implicit f.E::SrvBuf 'hlsl::StructuredBuffer<int>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "t0" "space0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 10 0 // CHECK: VarDecl {{.*}} implicit f.a.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u20" "space0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 11 0 // CHECK: VarDecl {{.*}} implicit f.SrvBuf 'hlsl::StructuredBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "t1" "space0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 12 0 // CHECK: VarDecl {{.*}} implicit f.Samp 'hlsl::SamplerState' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "s3" "space0" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 13 0 // CHECK: VarDecl {{.*}} f 'hlsl_constant F' -// CHECK: HLSLResourceBindingAttr {{.*}} "t0" "space0" -// CHECK: HLSLResourceBindingAttr {{.*}} "u20" "space0" -// CHECK: HLSLResourceBindingAttr {{.*}} "s3" "space0" -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.E::SrvBuf' 'hlsl::StructuredBuffer<int>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.a.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.SrvBuf' 'hlsl::StructuredBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'f.Samp' 'hlsl::SamplerState' +// CHECK-NEXT: HLSLVkBindingAttr {{.*}} 10 0 +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "t0" "space0" +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "u20" "space0" +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "s3" "space0" +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'f.E::SrvBuf' 'hlsl::StructuredBuffer<int>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'f.a.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'f.SrvBuf' 'hlsl::StructuredBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'f.Samp' 'hlsl::SamplerState' +[[vk::binding(10)]] F f : register(t0) : register(u20) : register(s3); // Array of structs with resources // CHECK: VarDecl {{.*}} implicit arrayOfA.0.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u0" "space1" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 0 1 // CHECK: VarDecl {{.*}} implicit arrayOfA.1.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u1" "space1" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 1 1 + +// CHECK: VarDecl {{.*}} arrayOfA 'hlsl_constant A[2]' +// CHECK-NEXT: HLSLVkBindingAttr {{.*}} 0 1 +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "u0" "space1" +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'arrayOfA.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'arrayOfA.1.Buf' 'hlsl::RWBuffer<float>' +[[vk::binding(0, 1)]] A arrayOfA[2] : register(u0, space1); // CHECK: CXXRecordDecl {{.*}} struct G @@ -126,38 +153,48 @@ struct G { }; // CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.0.0.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u10" "space2" +// SPRIV: HLSLVkBindingAttr {{.*}} Implicit 10 2 // CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.0.1.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u11" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 11 2 // CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.1.0.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" - +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u12" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 12 2 // CHECK: VarDecl {{.*}} implicit gArray.0.multiArray.1.1.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u13" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 13 2 // CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.0.0.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u14" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 14 2 // CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.0.1.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u15" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 15 2 // CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.1.0.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u16" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 16 2 // CHECK: VarDecl {{.*}} implicit gArray.1.multiArray.1.1.Buf 'hlsl::RWBuffer<float>' callinit -// CHECK: HLSLResourceBindingAttr {{.*}} Implicit "" "0" +// DXIL: HLSLResourceBindingAttr {{.*}} Implicit "u17" "space2" +// SPIRV: HLSLVkBindingAttr {{.*}} Implicit 17 2 // CHECK: VarDecl {{.*}} gArray 'hlsl_constant G[2]' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.0.0.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.0.1.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.1.0.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.1.1.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.0.0.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.0.1.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.1.0.Buf' 'hlsl::RWBuffer<float>' -// CHECK: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.1.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLVkBindingAttr {{.*}} 10 2 +// CHECK-NEXT: HLSLResourceBindingAttr {{.*}} "u10" "space2" +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.0.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.0.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.1.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.0.multiArray.1.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.0.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.0.1.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.1.0.Buf' 'hlsl::RWBuffer<float>' +// CHECK-NEXT: HLSLAssociatedResourceDeclAttr {{.*}} 'gArray.1.multiArray.1.1.Buf' 'hlsl::RWBuffer<float>' +[[vk::binding(10, 2)]] G gArray[2] : register(u10, space2); // Static struct with resources >From be5da9208e9ed5816db2298152a3aae03869c227 Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Mon, 9 Mar 2026 13:35:13 -0700 Subject: [PATCH 5/9] clang-format --- clang/lib/Sema/SemaHLSL.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index 0d1fc9d9cb7de..2f6530b3521ac 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -4843,7 +4843,7 @@ void SemaHLSL::ActOnVariableDeclarator(VarDecl *VD) { if (VD->getStorageClass() != SC_Static && VDTy->isHLSLIntangibleType() && !isResourceRecordTypeOrArrayOf(VD)) handleGlobalStructOrArrayOfWithResources(VD); - + // Mark groupshared variables as extern so they will have // external storage and won't be default initialized if (VD->hasAttr<HLSLGroupSharedAddressSpaceAttr>()) >From 29b26ecbdd9aa2f9bf5f2b46ba3a2258693964f1 Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Tue, 17 Mar 2026 14:23:24 -0700 Subject: [PATCH 6/9] [HLSL] Diagnose dynamic indexing of struct arrays for resource access --- .../clang/Basic/DiagnosticSemaKinds.td | 3 ++ clang/include/clang/Sema/SemaHLSL.h | 1 + clang/lib/Sema/SemaExprMember.cpp | 14 +++++- clang/lib/Sema/SemaHLSL.cpp | 46 +++++++++++++++++++ .../resources-in-struct-array-error.hlsl | 22 +++++++++ 5 files changed, 84 insertions(+), 2 deletions(-) create mode 100644 clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl diff --git a/clang/include/clang/Basic/DiagnosticSemaKinds.td b/clang/include/clang/Basic/DiagnosticSemaKinds.td index 0e6b3f51a5231..a3579d389fccc 100644 --- a/clang/include/clang/Basic/DiagnosticSemaKinds.td +++ b/clang/include/clang/Basic/DiagnosticSemaKinds.td @@ -13645,6 +13645,9 @@ def err_hlsl_samplecmp_requires_float def err_hlsl_gathercmp_invalid_component : Error<"gatherCmp%select{Red|Green|Blue|Alpha}0 operations on the Vulkan target are not supported; only GatherCmp and GatherCmpRed are allowed">; +def err_hlsl_resource_member_array_access_not_constant + : Error<"index for struct array inside cbuffer that contains resources must be a literal expression">; + // Layout randomization diagnostics. def err_non_designated_init_used : Error< "a randomized struct can only be initialized with a designated initializer">; diff --git a/clang/include/clang/Sema/SemaHLSL.h b/clang/include/clang/Sema/SemaHLSL.h index cf9b8c39b109e..4aa2f4a224bf0 100644 --- a/clang/include/clang/Sema/SemaHLSL.h +++ b/clang/include/clang/Sema/SemaHLSL.h @@ -131,6 +131,7 @@ class SemaHLSL : public SemaBase { void ActOnVariableDeclarator(VarDecl *VD); bool ActOnUninitializedVarDecl(VarDecl *D); void ActOnEndOfTranslationUnit(TranslationUnitDecl *TU); + bool ActOnResourceMemberAccessExpr(MemberExpr *ME); void CheckEntryPoint(FunctionDecl *FD); // Return true if everything is ok; returns false if there was an error. diff --git a/clang/lib/Sema/SemaExprMember.cpp b/clang/lib/Sema/SemaExprMember.cpp index ff7c844f66123..a4504410cae28 100644 --- a/clang/lib/Sema/SemaExprMember.cpp +++ b/clang/lib/Sema/SemaExprMember.cpp @@ -12,6 +12,7 @@ #include "clang/AST/DeclCXX.h" #include "clang/AST/DeclObjC.h" #include "clang/AST/DeclTemplate.h" +#include "clang/AST/Expr.h" #include "clang/AST/ExprCXX.h" #include "clang/AST/ExprObjC.h" #include "clang/AST/TypeBase.h" @@ -1737,9 +1738,18 @@ ExprResult Sema::ActOnMemberAccessExpr(Scope *S, Expr *Base, Base, Base->getType(), OpLoc, IsArrow, SS, TemplateKWLoc, FirstQualifierInScope, NameInfo, TemplateArgs, S, &ExtraArgs); - if (!Res.isInvalid() && isa<MemberExpr>(Res.get())) - CheckMemberAccessOfNoDeref(cast<MemberExpr>(Res.get())); + if (!Res.isInvalid()) { + if (MemberExpr *ME = dyn_cast<MemberExpr>(Res.get())) { + CheckMemberAccessOfNoDeref(ME); + if (getLangOpts().HLSL) { + QualType Ty = Res.get()->getType(); + if (Ty->isHLSLResourceRecord() || Ty->isHLSLResourceRecordArray()) + if (!HLSL().ActOnResourceMemberAccessExpr(ME)) + Res = ExprError(); + } + } + } return Res; } diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index 6ea8750482af7..bf0834ec3c120 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -2994,6 +2994,52 @@ void SemaHLSL::ActOnEndOfTranslationUnit(TranslationUnitDecl *TU) { diagnoseAvailabilityViolations(TU); } +// For resource member access through a global struct array, verify that the +// array index selecting the struct element is a constant integer expression. +// Returns false if the member expression is invalid. +bool SemaHLSL::ActOnResourceMemberAccessExpr(MemberExpr *ME) { + assert((ME->getType()->isHLSLResourceRecord() || + ME->getType()->isHLSLResourceRecordArray()) && + "expected member expr to have resource record type or array of them"); + + // Walk the AST from MemberExpr to the VarDecl of the parent struct instance + // and take note of any non-constant array indexing along the way. If the + // VarDecl we find is a global variable, report error if there was any + // non-constant array index in the resource member access along the way. + const Expr *NonConstIndexExpr = nullptr; + const Expr *E = ME->getBase(); + while (E) { + if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) { + if (!NonConstIndexExpr) + return true; + + assert(isa<VarDecl>(DRE->getDecl()) && + "member expr base is not a var decl"); + const VarDecl *VD = cast<VarDecl>(DRE->getDecl()); + if (!VD->hasGlobalStorage()) + return true; + + SemaRef.Diag(NonConstIndexExpr->getExprLoc(), + diag::err_hlsl_resource_member_array_access_not_constant); + return false; + } + + if (const auto *ASE = dyn_cast<ArraySubscriptExpr>(E)) { + const Expr *IdxExpr = ASE->getIdx(); + if (!IdxExpr->isIntegerConstantExpr(SemaRef.getASTContext())) + NonConstIndexExpr = IdxExpr; + E = ASE->getBase(); + } else if (const auto *SubME = dyn_cast<MemberExpr>(E)) { + E = SubME->getBase(); + } else if (const auto *ICE = dyn_cast<ImplicitCastExpr>(E)) { + E = ICE->getSubExpr(); + } else { + llvm_unreachable("unexpected expr type in resource member access"); + } + } + return true; +} + void SemaHLSL::diagnoseAvailabilityViolations(TranslationUnitDecl *TU) { // Skip running the diagnostics scan if the diagnostic mode is // strict (-fhlsl-strict-availability) and the target shader stage is known diff --git a/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl b/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl new file mode 100644 index 0000000000000..978592f36e6b8 --- /dev/null +++ b/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl @@ -0,0 +1,22 @@ +// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.0-compute -finclude-default-header -verify %s + +struct A { + RWBuffer<float> Buf; + RWBuffer<float> ManyBufs[5]; +}; + +A array[10] : register(u10); + +[numthreads(4,1,1)] +void main(uint GI : SV_GroupThreadID) { + + // expected-error@+1 {{index for struct array inside cbuffer that contains resources must be a literal expression}} + array[GI].Buf[0] = 1.0f; + + array[2].Buf[GI] = 2.0f; // ok + + // expected-error@+1 {{index for struct array inside cbuffer that contains resources must be a literal expression}} + array[GI].ManyBufs[3][0] = 3.0f; + + array[1].ManyBufs[GI][0] = 4.0f; // ok +} >From 4595004be090a03766a5b7aa624d92d2d5441321 Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Mon, 30 Mar 2026 16:34:01 -0700 Subject: [PATCH 7/9] core review feedback - remove redundant assert --- clang/lib/Sema/SemaHLSL.cpp | 2 -- 1 file changed, 2 deletions(-) diff --git a/clang/lib/Sema/SemaHLSL.cpp b/clang/lib/Sema/SemaHLSL.cpp index bf0834ec3c120..00ca2dbe2b695 100644 --- a/clang/lib/Sema/SemaHLSL.cpp +++ b/clang/lib/Sema/SemaHLSL.cpp @@ -3013,8 +3013,6 @@ bool SemaHLSL::ActOnResourceMemberAccessExpr(MemberExpr *ME) { if (!NonConstIndexExpr) return true; - assert(isa<VarDecl>(DRE->getDecl()) && - "member expr base is not a var decl"); const VarDecl *VD = cast<VarDecl>(DRE->getDecl()); if (!VD->hasGlobalStorage()) return true; >From cd51f45afdfb0d5407385235f8168c655a7b21ee Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Thu, 2 Apr 2026 18:10:15 -0700 Subject: [PATCH 8/9] code review feedback - update error message, add test --- clang/include/clang/Basic/DiagnosticSemaKinds.td | 2 +- .../SemaHLSL/Resources/resources-in-struct-array-error.hlsl | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/clang/include/clang/Basic/DiagnosticSemaKinds.td b/clang/include/clang/Basic/DiagnosticSemaKinds.td index d0e60a4e3728c..75728c5d42809 100644 --- a/clang/include/clang/Basic/DiagnosticSemaKinds.td +++ b/clang/include/clang/Basic/DiagnosticSemaKinds.td @@ -13700,7 +13700,7 @@ def err_hlsl_gathercmp_invalid_component : Error<"gatherCmp%select{Red|Green|Blue|Alpha}0 operations on the Vulkan target are not supported; only GatherCmp and GatherCmpRed are allowed">; def err_hlsl_resource_member_array_access_not_constant - : Error<"index for struct array inside cbuffer that contains resources must be a literal expression">; + : Error<"index for struct array inside cbuffer that contains resources must be a constant integer expression">; // Layout randomization diagnostics. def err_non_designated_init_used : Error< diff --git a/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl b/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl index 978592f36e6b8..1653d23c2c0ea 100644 --- a/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl +++ b/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl @@ -10,13 +10,15 @@ A array[10] : register(u10); [numthreads(4,1,1)] void main(uint GI : SV_GroupThreadID) { - // expected-error@+1 {{index for struct array inside cbuffer that contains resources must be a literal expression}} + // expected-error@+1 {{index for struct array inside cbuffer that contains resources must be a constant integer expression}} array[GI].Buf[0] = 1.0f; array[2].Buf[GI] = 2.0f; // ok - // expected-error@+1 {{index for struct array inside cbuffer that contains resources must be a literal expression}} + // expected-error@+1 {{index for struct array inside cbuffer that contains resources must be a constant integer expression}} array[GI].ManyBufs[3][0] = 3.0f; array[1].ManyBufs[GI][0] = 4.0f; // ok + + array[1+1].ManyBufs[GI][0] = 4.0f; // ok } >From 49c49532d6cb228c0aba5b6338ef72e0c5d8b4ce Mon Sep 17 00:00:00 2001 From: Helena Kotas <[email protected]> Date: Tue, 7 Apr 2026 17:16:25 -0700 Subject: [PATCH 9/9] code review feedback - add one more test case --- .../SemaHLSL/Resources/resources-in-struct-array-error.hlsl | 3 +++ 1 file changed, 3 insertions(+) diff --git a/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl b/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl index 1653d23c2c0ea..e98a73812b773 100644 --- a/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl +++ b/clang/test/SemaHLSL/Resources/resources-in-struct-array-error.hlsl @@ -21,4 +21,7 @@ void main(uint GI : SV_GroupThreadID) { array[1].ManyBufs[GI][0] = 4.0f; // ok array[1+1].ManyBufs[GI][0] = 4.0f; // ok + + int x = 3; + array[1].ManyBufs[x][0] = 4.0f; // ok } _______________________________________________ cfe-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
