fixup: Put cbuffer size in param

Author: bogner

llvm/include/llvm/Analysis/DXILResource.h

@@ -164,6 +164,7 @@ class CBufferExtType : public TargetExtType {
   CBufferExtType &operator=(const CBufferExtType &) = delete;
 
   Type *getResourceType() const { return getTypeParameter(0); }
+  uint32_t getCBufferSize() const { return getIntParameter(0); }
 
   static bool classof(const TargetExtType *T) {
     return T->getName() == "dx.CBuffer";

llvm/lib/Analysis/DXILResource.cpp

@@ -296,8 +296,7 @@ ResourceInfo::UAVInfo ResourceInfo::getUAV() const {
 
 uint32_t ResourceInfo::getCBufferSize(const DataLayout &DL) const {
   assert(isCBuffer() && "Not a CBuffer");
-  Type *Ty = cast<CBufferExtType>(HandleTy)->getResourceType();
-  return DL.getTypeSizeInBits(Ty) / 8;
+  return cast<CBufferExtType>(HandleTy)->getCBufferSize();
 }
 
 dxil::SamplerType ResourceInfo::getSamplerType() const {

llvm/unittests/Analysis/DXILResourceTest.cpp

@@ -380,7 +380,8 @@ TEST(DXILResource, AnnotationsAndMetadata) {
   // cbuffer cb0 { float4 g_X; float4 g_Y; }
   StructType *CBufType0 =
       StructType::create(Context, {Floatx4Ty, Floatx4Ty}, "cb0");
-  HandleTy = llvm::TargetExtType::get(Context, "dx.CBuffer", CBufType0, {});
+  HandleTy =
+      llvm::TargetExtType::get(Context, "dx.CBuffer", CBufType0, {/*Size=*/32});
   RI = ResourceInfo(
       /*RecordID=*/0, /*Space=*/0, /*LowerBound=*/0, /*Size=*/1, HandleTy);