Add dedicated CIR op for setjmp

Author: badumbatish

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

@@ -4698,6 +4698,19 @@ def CIR_EhTypeIdOp : CIR_Op<"eh.typeid",
   }];
 }
 
+def CIR_EhSjljSetjmp : CIR_Op<"eh.sjlj.setjmp"> {
+  let summary = "CIR setjmp operation";
+  let description = [{
+    "Dedicated setjmp operation for longjmp, a nonlocal jump"
+  }];
+
+  let arguments = (ins CIR_PointerType:$buf);
+  let results = (outs CIR_SInt32:$res);
+  let assemblyFormat = [{
+    $buf `:` type($buf) `->` type(results) attr-dict
+  }];
+}
+
 //===----------------------------------------------------------------------===//
 // CopyOp
 //===----------------------------------------------------------------------===//

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

@@ -4288,6 +4288,14 @@ mlir::LogicalResult CIRToLLVMEhTypeIdOpLowering::matchAndRewrite(
   return mlir::success();
 }
 
+mlir::LogicalResult CIRToLLVMEhSjljSetjmpOpLowering::matchAndRewrite(
+    cir::EhSjljSetjmp op, OpAdaptor adaptor,
+    mlir::ConversionPatternRewriter &rewriter) const {
+  rewriter.replaceOpWithNewOp<mlir::LLVM::Eh_SJLJ_Setjmp>(
+      op, typeConverter->convertType(op.getType()), adaptor.getBuf());
+  return mlir::success();
+}
+
 mlir::LogicalResult CIRToLLVMCatchParamOpLowering::matchAndRewrite(
     cir::CatchParamOp op, OpAdaptor adaptor,
     mlir::ConversionPatternRewriter &rewriter) const {
@@ -4531,6 +4539,7 @@ void populateCIRToLLVMConversionPatterns(
       CIRToLLVMDerivedClassAddrOpLowering,
       CIRToLLVMEhInflightOpLowering,
       CIRToLLVMEhTypeIdOpLowering,
+      CIRToLLVMEhSjljSetjmpOpLowering,
       CIRToLLVMExpectOpLowering,
       CIRToLLVMExtractMemberOpLowering,
       CIRToLLVMFrameAddrOpLowering,

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

@@ -1220,6 +1220,16 @@ class CIRToLLVMEhTypeIdOpLowering
                   mlir::ConversionPatternRewriter &) const override;
 };
 
+class CIRToLLVMEhSjljSetjmpOpLowering
+    : public mlir::OpConversionPattern<cir::EhSjljSetjmp> {
+public:
+  using mlir::OpConversionPattern<cir::EhSjljSetjmp>::OpConversionPattern;
+
+  mlir::LogicalResult
+  matchAndRewrite(cir::EhSjljSetjmp op, OpAdaptor,
+                  mlir::ConversionPatternRewriter &) const override;
+};
+
 class CIRToLLVMCatchParamOpLowering
     : public mlir::OpConversionPattern<cir::CatchParamOp> {
 public:

clang/test/CIR/Lowering/setjmp-longjmp.cir

@@ -0,0 +1,20 @@
+// RUN: cir-opt %s -cir-to-llvm -o - | FileCheck %s -check-prefix=MLIR
+
+!s32i = !cir.int<s, 32>
+!p32  = !cir.ptr<!s32i>
+
+module {
+  // MLIR: module {
+  cir.func @setjmp(%arg0 : !p32) -> !s32i {
+
+    // MLIR:  llvm.func @setjmp(%arg0: !llvm.ptr) -> i32
+    // MLIR-NEXT:    %0 = llvm.intr.eh.sjlj.setjmp %arg0 : (!llvm.ptr) -> i32
+    // MLIR-NEXT:    llvm.return %0 : i32
+    // MLIR-NEXT:  }
+
+    %0 = cir.eh.sjlj.setjmp %arg0 : !p32 -> !s32i
+    cir.return %0 : !s32i
+  }
+  // MLIR: }
+}
+

mlir/include/mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td

@@ -779,6 +779,12 @@ def LLVM_EhTypeidForOp : LLVM_OneResultIntrOp<"eh.typeid.for", [], [0]> {
     let assemblyFormat = "$type_info attr-dict `:` functional-type(operands, results)";
 }
 
+def LLVM_Eh_SJLJ_Setjmp : LLVM_OneResultIntrOp<"eh.sjlj.setjmp", [], [0]> {
+  let arguments = (ins LLVM_AnyPointer:$buf);
+  let results = (outs I32:$res);
+  let assemblyFormat = "$buf attr-dict `:` functional-type(operands, results)";
+}
+
 //
 // Stack save/restore intrinsics.
 //