Package ai.djl.python.engine

Class MpiEngineProvider

java.lang.Object

ai.djl.python.engine.PyEngineProvider

ai.djl.python.engine.MpiEngineProvider

All Implemented Interfaces:

ai.djl.engine.EngineProvider

Direct Known Subclasses:

MpiEngineProvider.DsEngineProvider

public class MpiEngineProvider
extends PyEngineProvider

DsEngineProvider is the DeepSpeed implementation of EngineProvider.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	MpiEngineProvider.DsEngineProvider	DsEngineProvider is the alias of MpiEngineProvider.

Field Summary

Fields inherited from class ai.djl.python.engine.PyEngineProvider

mpiMode

Constructor Summary

Constructors

Constructor	Description
MpiEngineProvider()	Constructs a new MpiEngineProvider instance.

Method Summary

Modifier and Type	Method	Description
String	getEngineName()
int	getEngineRank()

Methods inherited from class ai.djl.python.engine.PyEngineProvider

getEngine

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

MpiEngineProvider

public MpiEngineProvider()

Constructs a new MpiEngineProvider instance.

Method Details

getEngineName

public String getEngineName()

Specified by:

getEngineName in interface ai.djl.engine.EngineProvider

Overrides:

getEngineName in class PyEngineProvider

getEngineRank

public int getEngineRank()

Specified by:

getEngineRank in interface ai.djl.engine.EngineProvider

Overrides:

getEngineRank in class PyEngineProvider