[CIR] Make UndefAttr use AttrBuilderWithInferredContext (#1577)

Author: xlauko

clang/include/clang/CIR/Dialect/IR/CIRAttrs.td

@@ -246,6 +246,13 @@ def UndefAttr : CIR_Attr<"Undef", "undef", [TypedAttrInterface]> {
   }];
 
   let parameters = (ins AttributeSelfTypeParameter<"">:$type);
+
+  let builders = [
+    AttrBuilderWithInferredContext<(ins "mlir::Type":$type), [{
+      return $_get(type.getContext(), type);
+    }]>
+  ];
+
   let assemblyFormat = [{}];
 }
 

clang/lib/CIR/CodeGen/CIRGenExprConst.cpp

@@ -353,7 +353,7 @@ mlir::Attribute ConstantAggregateBuilder::buildFrom(
   ConstantAggregateBuilderUtils Utils(CGM);
 
   if (Elems.empty())
-    return cir::UndefAttr::get(CGM.getBuilder().getContext(), DesiredTy);
+    return cir::UndefAttr::get(DesiredTy);
 
   auto Offset = [&](size_t I) { return Offsets[I] - StartOffset; };
 

clang/lib/CIR/CodeGen/CIRGenModule.cpp

@@ -1413,7 +1413,7 @@ void CIRGenModule::emitGlobalVarDefinition(const clang::VarDecl *d,
        d->getType()->isCUDADeviceBuiltinTextureType());
   if (getLangOpts().CUDA &&
       (isCudaSharedVar || isCudaShadowVar || isCudaDeviceShadowVar))
-    init = UndefAttr::get(&getMLIRContext(), convertType(d->getType()));
+    init = cir::UndefAttr::get(convertType(d->getType()));
   else if (d->hasAttr<LoaderUninitializedAttr>())
     assert(0 && "not implemented");
   else if (!initExpr) {

clang/lib/CIR/Dialect/IR/CIRMemorySlot.cpp

@@ -43,8 +43,8 @@ llvm::SmallVector<MemorySlot> cir::AllocaOp::getPromotableSlots() {
 
 Value cir::AllocaOp::getDefaultValue(const MemorySlot &slot,
                                      OpBuilder &builder) {
-  return builder.create<cir::ConstantOp>(
-      getLoc(), slot.elemType, builder.getAttr<cir::UndefAttr>(slot.elemType));
+  return builder.create<cir::ConstantOp>(getLoc(), slot.elemType,
+                                         cir::UndefAttr::get(slot.elemType));
 }
 
 void cir::AllocaOp::handleBlockArgument(const MemorySlot &slot,