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 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