Utility Functions for NVSHMEM4Py

The following are nvshmem.core APIs that can be used as-is from their bindings

nvshmem.core.direct.my_pe() → int

Get the current Processing Element (PE) ID of this process.

Returns:

int: The PE ID of the calling process within TEAM_WORLD.

nvshmem.core.direct.team_my_pe(team) → int

Get the PE ID of this process within a specified team.

Args:

team: The team handle (e.g., nvshmem.core.Teams.TEAM_NODE).

Returns:

int: The PE ID of the calling process within the specified team.

nvshmem.core.direct.team_n_pes(team) → int

Get the number of Processing Elements (PEs) in a specified team.

Args:

team: The team handle (e.g., nvshmem.core.Teams.TEAM_NODE).

Returns:

int: The total number of PEs in the specified team.

nvshmem.core.direct.n_pes() → int

Get the total number of Processing Elements (PEs) in TEAM_WORLD.

Returns:

int: The total number of PEs in the default global team (TEAM_WORLD).

nvshmem.core.direct.team_n_pes(team) → int

Get the number of Processing Elements (PEs) in a specified team.

Args:

team: The team handle (e.g., nvshmem.core.Teams.TEAM_NODE).

Returns:

int: The total number of PEs in the specified team.