add lifetime op

Author: el-ev

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

@@ -5721,6 +5721,18 @@ def LinkerOptionsOp : CIR_Op<"linker_options"> {
   let hasVerifier = 1;
 }
 
+//===----------------------------------------------------------------------===//
+// LifetimeStartOp & LifetimeEndOp
+//===----------------------------------------------------------------------===//
+
+class CIR_LifetimeBaseOp<string mnemonic> : CIR_Op<mnemonic, []> {
+  let arguments = (ins I64Attr:$size, CIR_PointerType:$ptr);
+  let assemblyFormat = "$size `,` $ptr attr-dict `:` qualified(type($ptr))";
+}
+
+def LifetimeStartOp : CIR_LifetimeBaseOp<"lifetime.start">;
+def LifetimeEndOp : CIR_LifetimeBaseOp<"lifetime.end">;
+
 //===----------------------------------------------------------------------===//
 // Standard library function calls
 //===----------------------------------------------------------------------===//

clang/lib/CIR/CodeGen/CIRGenDecl.cpp

@@ -763,6 +763,27 @@ void CIRGenFunction::emitExprAsInit(const Expr *init, const ValueDecl *D,
   llvm_unreachable("bad evaluation kind");
 }
 
+mlir::Attribute CIRGenFunction::emitLifetimeStart(int64_t size, mlir::Value addr) {
+  assert(!cir::MissingFeatures::shouldEmitLifetimeMarkers());
+  if (!shouldEmitLifetimeMarkers)
+    return nullptr;
+  assert(cast<cir::PointerType>(addr.getType()).getAddrSpace() ==
+             CGM.getCIRAllocaAddressSpace() &&
+         "Pointer should be in alloca address space");
+  mlir::IntegerAttr sizeAttr = builder.getI64IntegerAttr(size);
+  builder.create<cir::LifetimeStartOp>(*currSrcLoc, sizeAttr, addr);
+  return sizeAttr;
+}
+
+void CIRGenFunction::emitLifetimeEnd(int64_t size, mlir::Value addr) {
+  assert(!cir::MissingFeatures::shouldEmitLifetimeMarkers());
+  assert(cast<cir::PointerType>(addr.getType()).getAddrSpace() ==
+             CGM.getCIRAllocaAddressSpace() &&
+         "Pointer should be in alloca address space");
+  mlir::IntegerAttr sizeAttr = builder.getI64IntegerAttr(size);
+  builder.create<cir::LifetimeEndOp>(*currSrcLoc, sizeAttr, addr);
+}
+
 void CIRGenFunction::emitDecl(const Decl &D) {
   switch (D.getKind()) {
   case Decl::ImplicitConceptSpecialization:

clang/lib/CIR/CodeGen/CIRGenExpr.cpp

@@ -2349,7 +2349,7 @@ LValue CIRGenFunction::emitMaterializeTemporaryExpr(
       break;
 
     case SD_FullExpression: {
-      if (!ShouldEmitLifetimeMarkers)
+      if (!shouldEmitLifetimeMarkers)
         break;
       assert(0 && "NYI");
       break;

clang/lib/CIR/CodeGen/CIRGenFunction.cpp

@@ -40,7 +40,7 @@ CIRGenFunction::CIRGenFunction(CIRGenModule &cgm, CIRGenBuilderTy &builder,
                                bool suppressNewContext)
     : CIRGenTypeCache(cgm), CGM{cgm}, builder(builder),
       SanOpts(cgm.getLangOpts().Sanitize), CurFPFeatures(cgm.getLangOpts()),
-      ShouldEmitLifetimeMarkers(false) {
+      shouldEmitLifetimeMarkers(false) {
   if (!suppressNewContext)
     cgm.getCXXABI().getMangleContext().startNewFunction();
   EHStack.setCGF(this);
@@ -709,7 +709,7 @@ cir::FuncOp CIRGenFunction::generateCode(clang::GlobalDecl gd, cir::FuncOp fn,
 
     // Initialize helper which will detect jumps which can cause invalid
     // lifetime markers.
-    if (ShouldEmitLifetimeMarkers)
+    if (shouldEmitLifetimeMarkers)
       assert(!cir::MissingFeatures::shouldEmitLifetimeMarkers() && "NYI");
   }
 

clang/lib/CIR/CodeGen/CIRGenFunction.h

@@ -510,7 +510,7 @@ class CIRGenFunction : public CIRGenTypeCache {
   SymTableTy symbolTable;
   /// True if we need to emit the life-time markers. This is initially set in
   /// the constructor, but could be overwrriten to true if this is a coroutine.
-  bool ShouldEmitLifetimeMarkers;
+  bool shouldEmitLifetimeMarkers;
 
   using DeclMapTy = llvm::DenseMap<const clang::Decl *, Address>;
   /// This keeps track of the CIR allocas or globals for local C
@@ -2490,6 +2490,9 @@ class CIRGenFunction : public CIRGenTypeCache {
 
   mlir::Value emitX86BuiltinExpr(unsigned BuiltinID, const CallExpr *E);
 
+  mlir::Attribute emitLifetimeStart(int64_t size, mlir::Value addr);
+  void emitLifetimeEnd(int64_t size, mlir::Value addr);
+
   /// CIR build helpers
   /// -----------------
 public: