class ctranslate2.MpiInfo

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

Inherits from: pybind11_builtins.pybind11_object


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.