mscclpp.language.utils

Classes

AlgoSpec(name, collective, nranks_per_node, ...)
ReplicationPolicy(*values)

class mscclpp.language.utils.AlgoSpec(name, collective, nranks_per_node, world_size, in_place, instances, protocol, instr_fusion=True, auto_sync=True, replication_policy=ReplicationPolicy.interleaved, reuse_resources=False, num_threads_per_block=1024, use_double_scratch_buffer=False, buffer_alignment=16, min_message_size=0, max_message_size=18446744073709551615, tags=<factory>)

Bases: object

Parameters:

• name (str)

• collective (Collective)

• nranks_per_node (int)

• world_size (int)

• in_place (bool)

• instances (int)

• protocol (str)

• instr_fusion (bool)

• auto_sync (bool)

• replication_policy (ReplicationPolicy)

• reuse_resources (bool)

• num_threads_per_block (int)

• use_double_scratch_buffer (bool)

• buffer_alignment (int)

• min_message_size (int)

• max_message_size (int)

• tags (dict)

class mscclpp.language.utils.ReplicationPolicy(*values)

Bases: Enum