heat._config

Everything you need to know about the configuration of Heat

Module Contents

class MPILibrary(*args, **kwds)

Bases: enum.Enum

Create a collection of name/value pairs.

Example enumeration:

>>> class Color(Enum):
...     RED = 1
...     BLUE = 2
...     GREEN = 3

Access them by:

• attribute access:

>>> Color.RED
<Color.RED: 1>

• value lookup:

>>> Color(1)
<Color.RED: 1>

• name lookup:

>>> Color['RED']
<Color.RED: 1>

Enumerations can be iterated over, and know how many members they have:

>>> len(Color)
3

>>> list(Color)
[<Color.RED: 1>, <Color.BLUE: 2>, <Color.GREEN: 3>]

Methods can be added to enumerations, and members can have their own attributes – see the documentation for details.

OpenMPI = 'ompi'

IntelMPI = 'impi'

MVAPICH = 'mvapich'

MPICH = 'mpich'

CrayMPI = 'craympi'

ParaStationMPI = 'psmpi'

Other = 'other'

class MPILibraryInfo

name :MPILibrary

version :str

_get_mpi_library() → MPILibraryInfo

_check_gpu_aware_mpi(library: MPILibraryInfo) → tuple[bool, bool]