[mlir] Add stacksave, stackrestore to llvm dialect

[ftynse]

mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td

1284

Nit: could we rather use LLVM_[Zero|One]ResultIntrOp instead of magic trailing parameters?

mlir/test/Target/llvmir-intrinsics.mlir

383

Looks like misrebase, I fixed this yesterday

[shabalin]

Address the feedback.

mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td

 }
 
 def LLVM_CoroResumeOp : LLVM_IntrOp<"coro.resume", [], [], [], 0> {
   let arguments = (ins LLVM_i8Ptr:$handle);
   let assemblyFormat = "$handle attr-dict";
 }
 
 //
+// Stack save/restore intrinsics.
+//
+
+def LLVM_StackSaveOp : LLVM_OneResultIntrOp<"stacksave"> {
+  let assemblyFormat = "attr-dict `:` type($res)";
+}
+
+def LLVM_StackRestoreOp : LLVM_ZeroResultIntrOp<"stackrestore"> {
+  let arguments = (ins LLVM_i8Ptr:$ptr);
+  let assemblyFormat = "$ptr attr-dict";
+}
+
+//
 // Vector Reductions.
 //
 
 def LLVM_vector_reduce_add : LLVM_VectorReduction<"add">;
 def LLVM_vector_reduce_and : LLVM_VectorReduction<"and">;
 def LLVM_vector_reduce_mul : LLVM_VectorReduction<"mul">;
 def LLVM_vector_reduce_fmax : LLVM_VectorReduction<"fmax">;
 def LLVM_vector_reduce_fmin : LLVM_VectorReduction<"fmin">;

mlir/test/Target/llvmir-intrinsics.mlir

llvm.func @coro_save(%arg0: !llvm.ptr<i8>) {
   // CHECK: call token @llvm.coro.save
   %0 = llvm.intr.coro.save %arg0 : !llvm.token
   llvm.return
 }
 
 // CHECK-LABEL: @coro_suspend
 llvm.func @coro_suspend(%arg0: !llvm.token, %arg1 : i1) {
   // CHECK: call i8 @llvm.coro.suspend
   %0 = llvm.intr.coro.suspend %arg0, %arg1 : i8
   llvm.return
 }
 
 // CHECK-LABEL: @coro_end
 llvm.func @coro_end(%arg0: !llvm.ptr<i8>, %arg1 : i1) {
   // CHECK: call i1 @llvm.coro.end
   %0 = llvm.intr.coro.end %arg0, %arg1 : i1
   llvm.return
 }
 
 // CHECK-LABEL: @coro_free
 llvm.func @coro_free(%arg0: !llvm.token, %arg1 : !llvm.ptr<i8>) {
   // CHECK: call i8* @llvm.coro.free
   %0 = llvm.intr.coro.free %arg0, %arg1 : !llvm.ptr<i8>
   llvm.return
 }
 
 // CHECK-LABEL: @coro_resume
 llvm.func @coro_resume(%arg0: !llvm.ptr<i8>) {
   // CHECK: call void @llvm.coro.resume
   llvm.intr.coro.resume %arg0
   llvm.return
 }
 
+// CHECK-LABEL: @stack_save
+llvm.func @stack_save() {
+  // CHECK: call i8* @llvm.stacksave
+  %0 = llvm.intr.stacksave : !llvm.ptr<i8>
+  llvm.return
+}
+
+// CHECK-LABEL: @stack_restore
+llvm.func @stack_restore(%arg0: !llvm.ptr<i8>) {
+  // CHECK: call void @llvm.stackrestore
+  llvm.intr.stackrestore %arg0
+  llvm.return
+}
+
 // Check that intrinsics are declared with appropriate types.
 // CHECK-DAG: declare float @llvm.fma.f32(float, float, float)
 // CHECK-DAG: declare <8 x float> @llvm.fma.v8f32(<8 x float>, <8 x float>, <8 x float>) #0
 // CHECK-DAG: declare float @llvm.fmuladd.f32(float, float, float)
 // CHECK-DAG: declare <8 x float> @llvm.fmuladd.v8f32(<8 x float>, <8 x float>, <8 x float>) #0
 // CHECK-DAG: declare void @llvm.prefetch.p0i8(i8* nocapture readonly, i32 immarg, i32 immarg, i32)
 // CHECK-DAG: declare float @llvm.exp.f32(float)
 // CHECK-DAG: declare <8 x float> @llvm.exp.v8f32(<8 x float>) #0