[Mlir-commits] [llvm] [mlir] [LLVM][NVPTX] Add Rubin extensions for G2S Tensor intrinsics (PR #220029)
Durgadoss R
llvmlistbot at llvm.org
Wed Sep 2 05:05:44 PDT 2026
================
@@ -1455,6 +1466,54 @@ class CP_ASYNC_BULK_TENSOR_PREFETCH_OVERRIDE_INTR<int dim, string mode,
[NVVM_ARG_NAME_PROP<!size(ArgsTy), "flag_cache_hint">.Prop]));
}
+class CP_ASYNC_BULK_TENSOR_G2S_OVERRIDE_INTR<int dim, string mode, string override>
+ : CP_ASYNC_BULK_TENSOR_OVERRIDE_BASE<dim, mode, override> {
+ string Name = Prefix # "g2s" # Suffix;
+ // The override_addr_dim/dim_stride variants support only 32-bit multicast masks.
+ LLVMType MulticastTy = !if(!eq(override, "override_addr"), llvm_anyint_ty,
+ llvm_i32_ty);
+ list<LLVMType> ArgsTy = !listconcat(
+ [llvm_shared_cluster_ptr_ty, // dst_smem_ptr
----------------
durga4github wrote:
nit: // dst_dsmem_ptr
https://github.com/llvm/llvm-project/pull/220029
More information about the Mlir-commits
mailing list