tvm_tirx.backend.cuda.op.nvshmem_n_pes

Contents

tvm_tirx.backend.cuda.op.nvshmem_n_pes#

tvm_tirx.backend.cuda.op.nvshmem_n_pes(*, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic to call nvshmem_n_pes()

Returns:

call (Expr) – The call expression.

Parameters:

loc (Location)