Shortcuts

Collective

class lightning_fabric.plugins.collectives.Collective[source]

Bases: abc.ABC

Interface for collective operations.

Supports communications between multiple processes and multiple nodes. A collective owns a group.

Warning

This API is experimental and subject to change

create_group(**kwargs)[source]

Create a group.

This assumes that init_group() has been called already by the user.

Return type

Self


© Copyright Copyright (c) 2018-2023, Lightning AI et al...

Built with Sphinx using a theme provided by Read the Docs.