Package ai.djl.python.engine

Class PyEngineProvider

java.lang.Object

ai.djl.python.engine.PyEngineProvider

All Implemented Interfaces:

ai.djl.engine.EngineProvider

Direct Known Subclasses:

MpiEngineProvider

public class PyEngineProvider
extends Object
implements ai.djl.engine.EngineProvider

PyEngineProvider is the Python implementation of EngineProvider.

Field Summary

Fields

Modifier and Type	Field	Description
protected boolean	mpiMode

Constructor Summary

Constructors

Constructor	Description
PyEngineProvider()

Method Summary

Modifier and Type	Method	Description
ai.djl.engine.Engine	getEngine()
String	getEngineName()
int	getEngineRank()

Methods inherited from class java.lang.Object

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

Field Details

mpiMode

protected boolean mpiMode

Constructor Details

PyEngineProvider

public PyEngineProvider()

Method Details

getEngineName

public String getEngineName()

Specified by:

getEngineName in interface ai.djl.engine.EngineProvider

getEngineRank

public int getEngineRank()

Specified by:

getEngineRank in interface ai.djl.engine.EngineProvider

getEngine

public ai.djl.engine.Engine getEngine()

Specified by:

getEngine in interface ai.djl.engine.EngineProvider