tvm_tirx.tirx.script.nvshmem.putmem_nbi.warp

Contents

tvm_tirx.tirx.script.nvshmem.putmem_nbi.warp#

tvm_tirx.tirx.script.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.

Contents