MpiInfo

class ctranslate2.MpiInfo

An object to manage the MPI communication between processes. It provides information about MPI connexion.

Inherits from: pybind11_builtins.pybind11_object

Methods:

• getCurRank

• getLocalRank

• getNRanks

static getCurRank() → int

Get the current rank of process.

static getLocalRank() → int

Get the current GPU id used by process.

static getNRanks() → int

Get the number of gpus running for the current model.