lower to llvm

Author: el-ev

clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp

@@ -4239,6 +4239,24 @@ mlir::LogicalResult CIRToLLVMLinkerOptionsOpLowering::matchAndRewrite(
   return mlir::success();
 }
 
+mlir::LogicalResult CIRToLLVMLifetimeStartOpLowering::matchAndRewrite(
+    cir::LifetimeStartOp op, OpAdaptor adaptor,
+    mlir::ConversionPatternRewriter &rewriter) const {
+  auto newOp = rewriter.create<mlir::LLVM::LifetimeStartOp>(
+      op.getLoc(), op.getSizeAttr(), adaptor.getPtr());
+  rewriter.replaceOp(op, newOp);
+  return mlir::success();
+}
+
+mlir::LogicalResult CIRToLLVMLifetimeEndOpLowering::matchAndRewrite(
+    cir::LifetimeEndOp op, OpAdaptor adaptor,
+    mlir::ConversionPatternRewriter &rewriter) const {
+  auto newOp = rewriter.create<mlir::LLVM::LifetimeEndOp>(
+      op.getLoc(), op.getSizeAttr(), adaptor.getPtr());
+  rewriter.replaceOp(op, newOp);
+  return mlir::success();
+}
+
 void populateCIRToLLVMConversionPatterns(
     mlir::RewritePatternSet &patterns, mlir::TypeConverter &converter,
     mlir::DataLayout &dataLayout, cir::LowerModule *lowerModule,
@@ -4323,6 +4341,8 @@ void populateCIRToLLVMConversionPatterns(
       CIRToLLVMInsertMemberOpLowering,
       CIRToLLVMIsConstantOpLowering,
       CIRToLLVMIsFPClassOpLowering,
+      CIRToLLVMLifetimeEndOpLowering,
+      CIRToLLVMLifetimeStartOpLowering,
       CIRToLLVMLinkerOptionsOpLowering,
       CIRToLLVMLLVMIntrinsicCallOpLowering,
       CIRToLLVMMemChrOpLowering,

clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.h

@@ -1257,6 +1257,24 @@ class CIRToLLVMLinkerOptionsOpLowering
                   mlir::ConversionPatternRewriter &rewriter) const override;
 };
 
+class CIRToLLVMLifetimeStartOpLowering
+    : public mlir::OpConversionPattern<cir::LifetimeStartOp> {
+public:
+  using mlir::OpConversionPattern<cir::LifetimeStartOp>::OpConversionPattern;
+  mlir::LogicalResult
+  matchAndRewrite(cir::LifetimeStartOp op, OpAdaptor,
+                  mlir::ConversionPatternRewriter &) const override;
+};
+
+class CIRToLLVMLifetimeEndOpLowering
+    : public mlir::OpConversionPattern<cir::LifetimeEndOp> {
+public:
+  using mlir::OpConversionPattern<cir::LifetimeEndOp>::OpConversionPattern;
+  mlir::LogicalResult
+  matchAndRewrite(cir::LifetimeEndOp op, OpAdaptor,
+                  mlir::ConversionPatternRewriter &) const override;
+};
+
 mlir::ArrayAttr lowerCIRTBAAAttr(mlir::Attribute tbaa,
                                  mlir::ConversionPatternRewriter &rewriter,
                                  cir::LowerModule *lowerMod);

clang/test/CIR/Lowering/lifetime.cir

@@ -0,0 +1,28 @@
+// RUN: cir-opt %s -cir-to-llvm -o - | FileCheck %s --check-prefix=MLIR
+// RUN: cir-translate %s -cir-to-llvmir --disable-cc-lowering  | FileCheck %s -check-prefix=LLVM
+
+!s32i = !cir.int<s, 32>
+
+module {
+  cir.func @foo(%arg0: !s32i) {
+    %0 = cir.alloca !s32i, !cir.ptr<!s32i>, %arg0 : !s32i, ["tmp"] {alignment = 16 : i64}
+    cir.lifetime.start 4, %0 : !cir.ptr<!s32i>
+    cir.lifetime.end 4, %0 : !cir.ptr<!s32i>
+    cir.return
+  }
+}
+
+//      MLIR: module {
+// MLIR-NEXT:  llvm.func @foo(%arg0: i32) attributes {cir.extra_attrs = #fn_attr, global_visibility = #cir<visibility default>} {
+// MLIR-NEXT:    %0 = llvm.alloca %arg0 x i32 {alignment = 16 : i64} : (i32) -> !llvm.ptr
+// MLIR-NEXT:    llvm.intr.lifetime.start 4, %0 : !llvm.ptr
+// MLIR-NEXT:    llvm.intr.lifetime.end 4, %0 : !llvm.ptr
+// MLIR-NEXT:    llvm.return
+// MLIR-NEXT:  }
+// MLIR-NEXT: }
+
+//      LLVM: define void @foo(i32 %0)
+// LLVM-NEXT:   %2 = alloca i32, i32 %0, align 16
+// LLVM-NEXT:   call void @llvm.lifetime.start.p0(i64 4, ptr %2)
+// LLVM-NEXT:   call void @llvm.lifetime.end.p0(i64 4, ptr %2)
+// LLVM-NEXT:   ret void