https://github.com/bob80905 updated https://github.com/llvm/llvm-project/pull/115971
>From 7413ceb21a6e0ac9212ef6317763ee0eff559612 Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Tue, 12 Nov 2024 16:44:00 -0800 Subject: [PATCH 1/7] print struct body within target ext ty context --- .../test/AST/HLSL/StructuredBuffer-AST-2.hlsl | 30 +++++++++++++++++++ llvm/lib/IR/AsmWriter.cpp | 10 +++++-- 2 files changed, 38 insertions(+), 2 deletions(-) create mode 100644 clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl diff --git a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl new file mode 100644 index 00000000000000..73073b3f6f2839 --- /dev/null +++ b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl @@ -0,0 +1,30 @@ +// RUN: %clang_dxc -T cs_6_6 %s | FileCheck %s + +// The purpose of this test is to ensure that the AST writer +// only emits struct bodies when within the context of a +// larger object that is being outputted on the RHS. + + +// note that "{ <4 x float> }" in the check below is a struct type, but only the +// body is emitted on the RHS because we are already in the context of a +// target extension type definition (class.hlsl::StructuredBuffer) +// CHECK: %"class.hlsl::StructuredBuffer" = type { target("dx.RawBuffer", { <4 x float> }, 0, 0), %struct.mystruct } +// CHECK: %struct.mystruct = type { <4 x float> } +// CHECK: %dx.types.Handle = type { ptr } +// CHECK: %dx.types.ResBind = type { i32, i32, i32, i8 } +// CHECK: %dx.types.ResourceProperties = type { i32, i32 } + +struct mystruct +{ + float4 Color; +}; + +StructuredBuffer<mystruct> my_buffer : register(t2, space4); + +export float4 test() +{ + return my_buffer[0].Color; +} + +[numthreads(1,1,1)] +void main() {} diff --git a/llvm/lib/IR/AsmWriter.cpp b/llvm/lib/IR/AsmWriter.cpp index 3c1cb76622bbb7..d2705ff4b30fb8 100644 --- a/llvm/lib/IR/AsmWriter.cpp +++ b/llvm/lib/IR/AsmWriter.cpp @@ -649,8 +649,14 @@ void TypePrinting::print(Type *Ty, raw_ostream &OS) { OS << "target(\""; printEscapedString(Ty->getTargetExtName(), OS); OS << "\""; - for (Type *Inner : TETy->type_params()) - OS << ", " << *Inner; + for (Type *Inner : TETy->type_params()) { + OS << ", "; + if (Inner->isStructTy()) { + StructType *STy = cast<StructType>(Inner); + printStructBody(STy, OS); + } else + OS << *Inner; + } for (unsigned IntParam : TETy->int_params()) OS << ", " << IntParam; OS << ")"; >From 69d90f218b9702f4bd194507e54b5af86e38d3b3 Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Wed, 13 Nov 2024 11:55:28 -0800 Subject: [PATCH 2/7] fix runline --- clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl index 73073b3f6f2839..6b71ddd567249f 100644 --- a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl +++ b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl @@ -1,4 +1,4 @@ -// RUN: %clang_dxc -T cs_6_6 %s | FileCheck %s +// RUN: %clang_cc1 -triple dxil-unknown-shadermodel6.6-compute -S -finclude-default-header -o - %s | FileCheck %s // The purpose of this test is to ensure that the AST writer // only emits struct bodies when within the context of a >From 3a1bdbacf556e2abffc2172721d66291e5f1e3ce Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Wed, 13 Nov 2024 13:54:54 -0800 Subject: [PATCH 3/7] undo change --- llvm/lib/IR/AsmWriter.cpp | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/llvm/lib/IR/AsmWriter.cpp b/llvm/lib/IR/AsmWriter.cpp index d2705ff4b30fb8..e1c8fa834984d9 100644 --- a/llvm/lib/IR/AsmWriter.cpp +++ b/llvm/lib/IR/AsmWriter.cpp @@ -649,14 +649,8 @@ void TypePrinting::print(Type *Ty, raw_ostream &OS) { OS << "target(\""; printEscapedString(Ty->getTargetExtName(), OS); OS << "\""; - for (Type *Inner : TETy->type_params()) { - OS << ", "; - if (Inner->isStructTy()) { - StructType *STy = cast<StructType>(Inner); - printStructBody(STy, OS); - } else - OS << *Inner; - } + for (Type *Inner : TETy->type_params()) + OS << ", " << *Inner; for (unsigned IntParam : TETy->int_params()) OS << ", " << IntParam; OS << ")"; >From c3e6120c0d3efb7b5aeae864e164976a1d869b6f Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Thu, 21 Nov 2024 17:08:57 -0800 Subject: [PATCH 4/7] convert named to anonymous structs when creating target ext types --- clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl | 9 +++++---- llvm/lib/IR/Type.cpp | 13 +++++++++++-- 2 files changed, 16 insertions(+), 6 deletions(-) diff --git a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl index 6b71ddd567249f..1edec7835f16c7 100644 --- a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl +++ b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl @@ -1,13 +1,14 @@ // RUN: %clang_cc1 -triple dxil-unknown-shadermodel6.6-compute -S -finclude-default-header -o - %s | FileCheck %s -// The purpose of this test is to ensure that the AST writer -// only emits struct bodies when within the context of a -// larger object that is being outputted on the RHS. - +// The purpose of this test is to ensure that the target +// extension type associated with the structured buffer only +// contains anonymous struct types, rather than named +// struct types // note that "{ <4 x float> }" in the check below is a struct type, but only the // body is emitted on the RHS because we are already in the context of a // target extension type definition (class.hlsl::StructuredBuffer) + // CHECK: %"class.hlsl::StructuredBuffer" = type { target("dx.RawBuffer", { <4 x float> }, 0, 0), %struct.mystruct } // CHECK: %struct.mystruct = type { <4 x float> } // CHECK: %dx.types.Handle = type { ptr } diff --git a/llvm/lib/IR/Type.cpp b/llvm/lib/IR/Type.cpp index 88ede0d35fa3ee..bc285fe31c44ca 100644 --- a/llvm/lib/IR/Type.cpp +++ b/llvm/lib/IR/Type.cpp @@ -794,8 +794,17 @@ TargetExtType::TargetExtType(LLVMContext &C, StringRef Name, // Parameter storage immediately follows the class in allocation. Type **Params = reinterpret_cast<Type **>(this + 1); ContainedTys = Params; - for (Type *T : Types) - *Params++ = T; + for (Type *T : Types) { + // target ext type parameters may not be named struct types + // so we should convert any named struct types to anonymous + // struct types in the parameter list + Type *ConvertedTy = T; + if (auto *STy = dyn_cast<StructType>(T)) { + if (STy->hasName()) + ConvertedTy = StructType::get(C, STy->elements(), /*isPacked=*/false); + } + *Params++ = ConvertedTy; + } setSubclassData(Ints.size()); unsigned *IntParamSpace = reinterpret_cast<unsigned *>(Params); >From 0d2579a66d3374114b0e0df4bab21d30f04a23c6 Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Thu, 21 Nov 2024 17:16:43 -0800 Subject: [PATCH 5/7] clang-format --- llvm/lib/IR/AsmWriter.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llvm/lib/IR/AsmWriter.cpp b/llvm/lib/IR/AsmWriter.cpp index e1c8fa834984d9..3c1cb76622bbb7 100644 --- a/llvm/lib/IR/AsmWriter.cpp +++ b/llvm/lib/IR/AsmWriter.cpp @@ -649,7 +649,7 @@ void TypePrinting::print(Type *Ty, raw_ostream &OS) { OS << "target(\""; printEscapedString(Ty->getTargetExtName(), OS); OS << "\""; - for (Type *Inner : TETy->type_params()) + for (Type *Inner : TETy->type_params()) OS << ", " << *Inner; for (unsigned IntParam : TETy->int_params()) OS << ", " << IntParam; >From b24c6d7188db07f5e37d818f894b43a0e50bb9fb Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Thu, 21 Nov 2024 17:57:08 -0800 Subject: [PATCH 6/7] add assert, move conversion --- clang/lib/CodeGen/Targets/DirectX.cpp | 13 +++++++++++-- llvm/lib/IR/Type.cpp | 8 ++++---- 2 files changed, 15 insertions(+), 6 deletions(-) diff --git a/clang/lib/CodeGen/Targets/DirectX.cpp b/clang/lib/CodeGen/Targets/DirectX.cpp index 7935f7ae370047..afa534ba2d0e47 100644 --- a/clang/lib/CodeGen/Targets/DirectX.cpp +++ b/clang/lib/CodeGen/Targets/DirectX.cpp @@ -43,9 +43,18 @@ llvm::Type *DirectXTargetCodeGenInfo::getHLSLType(CodeGenModule &CGM, if (ContainedTy.isNull()) return nullptr; - // convert element type llvm::Type *ElemType = CGM.getTypes().ConvertType(ContainedTy); + // target ext type parameters may not be named struct types + // so we should convert any named struct types to anonymous + // struct types in the parameter list + llvm::Type *ConvertedElemType = ElemType; + if (auto *STy = dyn_cast<llvm::StructType>(ElemType)) { + if (STy->hasName()) + ConvertedElemType = + llvm::StructType::get(Ctx, STy->elements(), /*isPacked=*/false); + } + llvm::StringRef TypeName = ResAttrs.RawBuffer ? "dx.RawBuffer" : "dx.TypedBuffer"; SmallVector<unsigned, 3> Ints = {/*IsWriteable*/ ResAttrs.ResourceClass == @@ -54,7 +63,7 @@ llvm::Type *DirectXTargetCodeGenInfo::getHLSLType(CodeGenModule &CGM, if (!ResAttrs.RawBuffer) Ints.push_back(/*IsSigned*/ ContainedTy->isSignedIntegerType()); - return llvm::TargetExtType::get(Ctx, TypeName, {ElemType}, Ints); + return llvm::TargetExtType::get(Ctx, TypeName, {ConvertedElemType}, Ints); } case llvm::dxil::ResourceClass::CBuffer: llvm_unreachable("dx.CBuffer handles are not implemented yet"); diff --git a/llvm/lib/IR/Type.cpp b/llvm/lib/IR/Type.cpp index bc285fe31c44ca..c9795fa335795c 100644 --- a/llvm/lib/IR/Type.cpp +++ b/llvm/lib/IR/Type.cpp @@ -799,10 +799,10 @@ TargetExtType::TargetExtType(LLVMContext &C, StringRef Name, // so we should convert any named struct types to anonymous // struct types in the parameter list Type *ConvertedTy = T; - if (auto *STy = dyn_cast<StructType>(T)) { - if (STy->hasName()) - ConvertedTy = StructType::get(C, STy->elements(), /*isPacked=*/false); - } + if (auto *STy = dyn_cast<StructType>(T)) + assert(!STy->hasName() && + "named structs are not allowed in target extension types"); + *Params++ = ConvertedTy; } >From e3f9344682659b718b2fb992d7781d852c5eb50c Mon Sep 17 00:00:00 2001 From: Joshua Batista <jbati...@microsoft.com> Date: Fri, 22 Nov 2024 10:21:09 -0800 Subject: [PATCH 7/7] change asmwriter --- clang/lib/CodeGen/Targets/DirectX.cpp | 12 +----------- clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl | 4 ++-- llvm/lib/IR/AsmWriter.cpp | 6 ++++-- llvm/lib/IR/Type.cpp | 13 ++----------- 4 files changed, 9 insertions(+), 26 deletions(-) diff --git a/clang/lib/CodeGen/Targets/DirectX.cpp b/clang/lib/CodeGen/Targets/DirectX.cpp index afa534ba2d0e47..257cf0ee8822fd 100644 --- a/clang/lib/CodeGen/Targets/DirectX.cpp +++ b/clang/lib/CodeGen/Targets/DirectX.cpp @@ -45,16 +45,6 @@ llvm::Type *DirectXTargetCodeGenInfo::getHLSLType(CodeGenModule &CGM, llvm::Type *ElemType = CGM.getTypes().ConvertType(ContainedTy); - // target ext type parameters may not be named struct types - // so we should convert any named struct types to anonymous - // struct types in the parameter list - llvm::Type *ConvertedElemType = ElemType; - if (auto *STy = dyn_cast<llvm::StructType>(ElemType)) { - if (STy->hasName()) - ConvertedElemType = - llvm::StructType::get(Ctx, STy->elements(), /*isPacked=*/false); - } - llvm::StringRef TypeName = ResAttrs.RawBuffer ? "dx.RawBuffer" : "dx.TypedBuffer"; SmallVector<unsigned, 3> Ints = {/*IsWriteable*/ ResAttrs.ResourceClass == @@ -63,7 +53,7 @@ llvm::Type *DirectXTargetCodeGenInfo::getHLSLType(CodeGenModule &CGM, if (!ResAttrs.RawBuffer) Ints.push_back(/*IsSigned*/ ContainedTy->isSignedIntegerType()); - return llvm::TargetExtType::get(Ctx, TypeName, {ConvertedElemType}, Ints); + return llvm::TargetExtType::get(Ctx, TypeName, {ElemType}, Ints); } case llvm::dxil::ResourceClass::CBuffer: llvm_unreachable("dx.CBuffer handles are not implemented yet"); diff --git a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl index 1edec7835f16c7..a45995bc6e34ea 100644 --- a/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl +++ b/clang/test/AST/HLSL/StructuredBuffer-AST-2.hlsl @@ -1,4 +1,4 @@ -// RUN: %clang_cc1 -triple dxil-unknown-shadermodel6.6-compute -S -finclude-default-header -o - %s | FileCheck %s +// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.6-compute -S -finclude-default-header -o - %s | FileCheck %s // The purpose of this test is to ensure that the target // extension type associated with the structured buffer only @@ -9,7 +9,7 @@ // body is emitted on the RHS because we are already in the context of a // target extension type definition (class.hlsl::StructuredBuffer) -// CHECK: %"class.hlsl::StructuredBuffer" = type { target("dx.RawBuffer", { <4 x float> }, 0, 0), %struct.mystruct } +// CHECK: %"class.hlsl::StructuredBuffer" = type { target("dx.RawBuffer", %struct.mystruct, 0, 0), %struct.mystruct } // CHECK: %struct.mystruct = type { <4 x float> } // CHECK: %dx.types.Handle = type { ptr } // CHECK: %dx.types.ResBind = type { i32, i32, i32, i8 } diff --git a/llvm/lib/IR/AsmWriter.cpp b/llvm/lib/IR/AsmWriter.cpp index 3c1cb76622bbb7..f8e17ac7bd6567 100644 --- a/llvm/lib/IR/AsmWriter.cpp +++ b/llvm/lib/IR/AsmWriter.cpp @@ -649,8 +649,10 @@ void TypePrinting::print(Type *Ty, raw_ostream &OS) { OS << "target(\""; printEscapedString(Ty->getTargetExtName(), OS); OS << "\""; - for (Type *Inner : TETy->type_params()) - OS << ", " << *Inner; + for (Type *Inner : TETy->type_params()) { + OS << ", "; + Inner->print(OS, /*IsForDebug=*/false, /*NoDetails=*/true); + } for (unsigned IntParam : TETy->int_params()) OS << ", " << IntParam; OS << ")"; diff --git a/llvm/lib/IR/Type.cpp b/llvm/lib/IR/Type.cpp index c9795fa335795c..88ede0d35fa3ee 100644 --- a/llvm/lib/IR/Type.cpp +++ b/llvm/lib/IR/Type.cpp @@ -794,17 +794,8 @@ TargetExtType::TargetExtType(LLVMContext &C, StringRef Name, // Parameter storage immediately follows the class in allocation. Type **Params = reinterpret_cast<Type **>(this + 1); ContainedTys = Params; - for (Type *T : Types) { - // target ext type parameters may not be named struct types - // so we should convert any named struct types to anonymous - // struct types in the parameter list - Type *ConvertedTy = T; - if (auto *STy = dyn_cast<StructType>(T)) - assert(!STy->hasName() && - "named structs are not allowed in target extension types"); - - *Params++ = ConvertedTy; - } + for (Type *T : Types) + *Params++ = T; setSubclassData(Ints.size()); unsigned *IntParamSpace = reinterpret_cast<unsigned *>(Params); _______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits