tvm_tirx.tirx.script.nvshmem.n_pes

tvm_tirx.tirx.script.nvshmem.n_pes(*, ty=None, loc=UNKNOWN_LOC)

TVM intrinsic to call nvshmem_n_pes()

Returns:

call (Expr) – The call expression.

Parameters:

loc (Location)