[Mlir-commits] [mlir] [MLIR][NVVM] Add support for st.bulk Op (PR #131727)

[Srinivasa Ravi]

================
@@ -2583,6 +2583,25 @@ def NVVM_MapaOp: NVVM_Op<"mapa",
   let assemblyFormat = "$a`,` $b attr-dict `:` type($a) `->` type($res)";
 }
 
+//===----------------------------------------------------------------------===//
+// NVVM Bulk Store Op
+//===----------------------------------------------------------------------===//
+
+def NVVM_BulkStoreOp: NVVM_Op<"st.bulk"> {
+  let arguments = (ins AnyTypeOf<[LLVM_PointerGeneric, LLVM_PointerShared]>:$addr, I64:$size, I64Attr:$initval);
+
+  string llvmBuilder = [{
+    int addrSpace = llvm::cast<LLVMPointerType>(op.getAddr().getType()).getAddressSpace();
----------------
Wolfram70 wrote:

I've moved the intrinsic choosing portion to `NVVMToLLVMIRTranslation.cpp` in the latest revision. Thanks!

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