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. var chunkedPackMemoryResource: Option[RmmPoolMemoryResource[RmmCudaMemoryResource]]
  6. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  7. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  8. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  9. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  10. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  11. def getDeviceId(): Option[Int]

    Exposes the device id used while initializing the RMM pool

  12. def getGPUAddrFromResources(resources: Map[String, ResourceInformation], conf: RapidsConf): Option[Int]
  13. def getNumCores: Int

    Get an approximate count on the number of cores this executor will use.

  14. def getResourcesFromTaskContext: Map[String, ResourceInformation]
  15. def getSplitUntilSize: Long
  16. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  17. 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 GPU per executor.

  18. def initializeGpu(resources: Map[String, ResourceInformation], conf: RapidsConf): Option[Int]
  19. def initializeGpuAndMemory(resources: Map[String, ResourceInformation], conf: RapidsConf, numCores: Int): Unit
  20. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  21. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  22. 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.

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