tvm_tirx.tirx.script.nvshmem.getmem_nbi

Contents

tvm_tirx.tirx.script.nvshmem.getmem_nbi#

Public canonical TVMScript dialect namespace.

Functions#

warp(src, nelems, pe, *[, ty, loc])

TVM intrinsic to call nvshmem_getmem_nbi_warp()

block(src, nelems, pe, *[, ty, loc])

TVM intrinsic to call nvshmem_getmem_nbi_block()