Skip to content
This repository has been archived by the owner on Nov 25, 2024. It is now read-only.

Mnnvl with split comm #185

Merged
merged 6 commits into from
Jun 18, 2024

Conversation

chuangz0
Copy link
Contributor

support split comm and get_local_mnnvl_comm

split_comm


def split_communicator(comm: WholeMemoryCommunicator, color: int, key: int = 0):
    """Split Communicator.
    Creates a set of new communicators from an existing one. Ranks which pass the same color value will be part of the
    same group; color must be a non-negative value.
    The value of key will determine the rank order, and the smaller key means the smaller rank in new communicator.
    If keys are equal between ranks, then the rank in the original communicator will be used to order ranks.
    """

@chuangz0 chuangz0 added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Jun 11, 2024
@chuangz0 chuangz0 marked this pull request as ready for review June 11, 2024 02:43
Copy link
Contributor

@linhu-nv linhu-nv left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Seems good to me.

@BradReesWork
Copy link
Member

/merge

@rapids-bot rapids-bot bot merged commit ba505af into rapidsai:branch-24.08 Jun 18, 2024
48 checks passed
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
improvement Improves an existing functionality non-breaking Introduces a non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants