[Mlir-commits] [llvm] [mlir] [LLVM][NVPTX][MLIR] Add mbarrier layout support (PR #217252)

[Rajat Bajpai]

================
@@ -2157,9 +2157,23 @@ foreach is_shared = [true, false] in {
   }
 }
 
+def int_nvvm_mbarrier_init_layout :
+  DefaultAttrsIntrinsic<[], [llvm_shared_ptr_ty,  // mbar
+                              llvm_i32_ty,        // count
+                              llvm_i32_ty],       // layout
+                        [IntrConvergent, IntrNoCallback, ImmArg<ArgIndex<2>>,
+                          Range<ArgIndex<2>, 0, 2>]>;
+
 def int_nvvm_mbarrier_pending_count : NVVMBuiltin,
     NVVMPureIntrinsic<[llvm_i32_ty], [llvm_i64_ty]>;
 
+def int_nvvm_mbarrier_check_layout :
+    DefaultAttrsIntrinsic<[llvm_i1_ty],
----------------
rajatbajpai wrote:

nit: is it possible to use `DefaultAttrsIntrinsicFlags`?

https://github.com/llvm/llvm-project/pull/217252