Packages

o

com.nvidia.spark.rapids

GpuDeviceManager

object GpuDeviceManager extends Logging

Linear Supertypes
Logging, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. GpuDeviceManager
  2. Logging
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. def clone(): AnyRef
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  6. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  10. def getDeviceId(): Option[Int]

    Exposes the device id used while initializing the RMM pool

  11. def getGPUAddrFromResources(resources: Map[String, ResourceInformation]): Option[Int]
  12. def getResourcesFromTaskContext: Map[String, ResourceInformation]
  13. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  14. def initializeFromTask(): Unit

    Always set the GPU if it was assigned by Spark and initialize the RMM if its configured to do so in the task.

    Always set the GPU if it was assigned by Spark and initialize the RMM if its configured to do so in the task. We expect the plugin to be run with 1 task and 1 GPU per executor.

  15. def initializeGpu(resources: Map[String, ResourceInformation]): Option[Int]
  16. def initializeGpuAndMemory(resources: Map[String, ResourceInformation]): Unit
  17. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  18. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  19. def initializeMemory(gpuId: Option[Int], rapidsConf: Option[RapidsConf] = None): Unit

    Initialize the GPU memory for gpuId according to the settings in rapidsConf.

    Initialize the GPU memory for gpuId according to the settings in rapidsConf. It is assumed that if gpuId is set then that gpu is already the default device. If gpuId is not set this will search all available GPUs starting at 0 looking for the appropriate one.

    gpuId

    the id of the gpu to use

    rapidsConf

    the config to use.

  20. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  21. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  22. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  23. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  24. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  25. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  26. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  27. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  28. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  29. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  30. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  31. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  32. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  33. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  34. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  35. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  36. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  37. var rmmTaskInitEnabled: Boolean
  38. def setGpuDeviceAndAcquire(addr: Int): Int
  39. def setRmmTaskInitEnabled(enabled: Boolean): Unit
  40. def shutdown(): Unit
  41. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  42. def toString(): String
    Definition Classes
    AnyRef → Any
  43. def tryToSetGpuDeviceAndAcquire(addr: Int): Boolean
  44. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  45. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  46. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  47. def wrapThreadFactory(factory: ThreadFactory): ThreadFactory

    Wrap a thread factory with one that will set the GPU device on each thread created.

Inherited from Logging

Inherited from AnyRef

Inherited from Any

Ungrouped