tvm_tirx.backend.cuda.op.nvshmem_putmem_nbi_warp

tvm_tirx.backend.cuda.op.nvshmem_putmem_nbi_warp(dst, src, nelems, pe, *, ty=None, loc=UNKNOWN_LOC)

TVM intrinsic to call nvshmem_putmem_nbi_warp()

Parameters:

• dst (Expr) – The pointer to the symmetric address of the destination data object.

• src (Expr) – The pointer to the symmetric address or host/device address of the data object to be copied.

• nelems (int) – The number of bytes to put per warp.

• pe (int) – The PE number of the remote PE.

• loc (Location)

Returns:

call (Expr) – The call expression.