tvm_tirx.tirx.script.nvshmem.putmem_nbi.block

Contents

tvm_tirx.tirx.script.nvshmem.putmem_nbi.block#

tvm_tirx.tirx.script.nvshmem.putmem_nbi.block(dst, src, nelems, pe, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic to call nvshmem_putmem_nbi_block()

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 block.

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

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents