tvm_tirx.backend.cuda.op.nvshmem_getmem_nbi

tvm_tirx.backend.cuda.op.nvshmem_getmem_nbi#

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

TVM intrinsic to call nvshmem_getmem_nbi()

Parameters:

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

  • src (Expr) – The pointer to the symmetric address of the source data object.

  • nelems (int) – The number of bytes to get per thread.

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

  • loc (Location)

Returns:

call (Expr) – The call expression.