Packages

c

com.nvidia.spark.rapids

MultiFileCloudPartitionReaderBase

abstract class MultiFileCloudPartitionReaderBase extends FilePartitionReaderBase

The Abstract multi-file cloud reading framework

The data driven: next() -> if (first time) initAndStartReaders -> submit tasks (getBatchRunner) -> wait tasks done sequentially -> decode in GPU (readBatch)

Linear Supertypes
FilePartitionReaderBase, ScanWithMetrics, Logging, PartitionReader[ColumnarBatch], Closeable, AutoCloseable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. MultiFileCloudPartitionReaderBase
  2. FilePartitionReaderBase
  3. ScanWithMetrics
  4. Logging
  5. PartitionReader
  6. Closeable
  7. AutoCloseable
  8. AnyRef
  9. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new MultiFileCloudPartitionReaderBase(conf: Configuration, inputFiles: Array[PartitionedFile], numThreads: Int, maxNumFileProcessed: Int, filters: Array[Filter], execMetrics: Map[String, GpuMetric], maxReadBatchSizeRows: Int, maxReadBatchSizeBytes: Long, ignoreCorruptFiles: Boolean = false, alluxioPathReplacementMap: Map[String, String] = Map.empty, alluxioReplacementTaskTime: Boolean = false, keepReadsInOrder: Boolean = true, combineConf: CombineConf = CombineConf(-1, -1))

    conf

    Configuration parameters

    inputFiles

    PartitionFiles to be read

    numThreads

    the number of threads to read files in parallel.

    maxNumFileProcessed

    threshold to control the maximum file number to be submitted to threadpool

    filters

    push down filters

    execMetrics

    the metrics

    ignoreCorruptFiles

    Whether to ignore corrupt files when GPU failed to decode the files

    alluxioPathReplacementMap

    Map containing mapping of DFS scheme to Alluxio scheme

    alluxioReplacementTaskTime

    Whether the Alluxio replacement algorithm is set to task time

    keepReadsInOrder

    Whether to require the files to be read in the same order as Spark. Defaults to true for formats that don't explicitly handle this.

    combineConf

    configs relevant to combination

Abstract Value Members

  1. abstract def getBatchRunner(tc: TaskContext, file: PartitionedFile, origFile: Option[PartitionedFile], conf: Configuration, filters: Array[Filter]): Callable[HostMemoryBuffersWithMetaDataBase]

    The sub-class must implement the real file reading logic in a Callable which will be running in a thread pool

    The sub-class must implement the real file reading logic in a Callable which will be running in a thread pool

    tc

    task context to use

    file

    file to be read

    origFile

    optional original unmodified file if replaced with Alluxio

    conf

    the Configuration parameters

    filters

    push down filters

    returns

    Callable[HostMemoryBuffersWithMetaDataBase]

  2. abstract def getFileFormatShortName: String

    File format short name used for logging and other things to uniquely identity which file format is being used.

    File format short name used for logging and other things to uniquely identity which file format is being used.

    returns

    the file format short name

  3. abstract def readBatches(fileBufsAndMeta: HostMemoryBuffersWithMetaDataBase): Iterator[ColumnarBatch]

    Decode HostMemoryBuffers in GPU

    Decode HostMemoryBuffers in GPU

    fileBufsAndMeta

    the file HostMemoryBuffer read from a PartitionedFile

    returns

    an iterator of batches that were decoded

Concrete 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 batchIter: Iterator[ColumnarBatch]
    Attributes
    protected
    Definition Classes
    FilePartitionReaderBase
  6. def canUseCombine: Boolean
  7. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  8. def close(): Unit
    Definition Classes
    MultiFileCloudPartitionReaderBaseFilePartitionReaderBase → Closeable → AutoCloseable
  9. def combineHMBs(results: Array[HostMemoryBuffersWithMetaDataBase]): HostMemoryBuffersWithMetaDataBase
  10. var combineLeftOverFiles: Option[Array[HostMemoryBuffersWithMetaDataBase]]
    Attributes
    protected
  11. var currentFileHostBuffers: Option[HostMemoryBuffersWithMetaDataBase]
    Attributes
    protected
  12. def currentMetricsValues(): Array[CustomTaskMetric]
    Definition Classes
    PartitionReader
  13. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  14. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  15. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  16. def get(): ColumnarBatch
    Definition Classes
    FilePartitionReaderBase → PartitionReader
  17. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  18. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  19. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  20. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  21. var isDone: Boolean
    Attributes
    protected
    Definition Classes
    FilePartitionReaderBase
  22. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  23. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  24. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  25. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  26. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  27. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  28. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  29. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  30. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  31. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  32. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  33. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  34. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  35. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  36. val metrics: Map[String, GpuMetric]
    Definition Classes
    ScanWithMetrics
  37. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  38. def next(): Boolean
    Definition Classes
    MultiFileCloudPartitionReaderBase → PartitionReader
  39. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  40. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  41. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  42. def toString(): String
    Definition Classes
    AnyRef → Any
  43. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  44. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  45. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from FilePartitionReaderBase

Inherited from ScanWithMetrics

Inherited from Logging

Inherited from PartitionReader[ColumnarBatch]

Inherited from Closeable

Inherited from AutoCloseable

Inherited from AnyRef

Inherited from Any

Ungrouped