tvm_tirx.tirx.script.nvshmem

tvm_tirx.tirx.script.nvshmem#

Public canonical TVMScript dialect namespace.

Functions#

my_pe(*[, loc])

TVM intrinsic to call nvshmem_my_pe()

n_pes(*[, loc])

TVM intrinsic to call nvshmem_n_pes()

signal_op(signal, sig_op, pe, *[, ty, loc])

TVM intrinsic to call nvshmem_signal_op()

wait_until(cmp, cmp_value[, type, ty, loc])

TVM intrinsic to call nvshmem_wait_until()

quiet(*[, loc])

TVM intrinsic to call nvshmem_quiet()

fence(*[, loc])

TVM intrinsic to call nvshmem_fence()

barrier_all(*[, loc])

TVM intrinsic to call nvshmem_barrier_all()

getmem_nbi

putmem_nbi

putmem_signal_nbi

Namespaces#