Packages

c

org.apache.spark.sql.rapids.execution

GpuCustomShuffleReaderExec

case class GpuCustomShuffleReaderExec(child: SparkPlan, partitionSpecs: Seq[ShufflePartitionSpec]) extends SparkPlan with ShimUnaryExecNode with GpuExec with Product with Serializable

A wrapper of shuffle query stage, which follows the given partition arrangement.

child

It is usually ShuffleQueryStageExec, but can be the shuffle exchange node during canonicalization.

partitionSpecs

The partition specs that defines the arrangement.

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. GpuCustomShuffleReaderExec
  2. GpuExec
  3. ShimUnaryExecNode
  4. UnaryExecNode
  5. SparkPlan
  6. Serializable
  7. Serializable
  8. Logging
  9. QueryPlan
  10. SQLConfHelper
  11. TreeNode
  12. Product
  13. Equals
  14. AnyRef
  15. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new GpuCustomShuffleReaderExec(child: SparkPlan, partitionSpecs: Seq[ShufflePartitionSpec])

    child

    It is usually ShuffleQueryStageExec, but can be the shuffle exchange node during canonicalization.

    partitionSpecs

    The partition specs that defines the arrangement.

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. lazy val additionalMetrics: Map[String, GpuMetric]
    Definition Classes
    GpuExec
  5. lazy val allAttributes: AttributeSeq
    Definition Classes
    QueryPlan
  6. lazy val allMetrics: Map[String, GpuMetric]

    We intentionally override metrics in this case rather than overriding additionalMetrics so that NUM_OUTPUT_ROWS and NUM_OUTPUT_BATCHES are removed, since this operator does not report any data for those metrics.

    We intentionally override metrics in this case rather than overriding additionalMetrics so that NUM_OUTPUT_ROWS and NUM_OUTPUT_BATCHES are removed, since this operator does not report any data for those metrics.

    The Spark version of this operator does not output any metrics.

    Definition Classes
    GpuCustomShuffleReaderExecGpuExec
  7. def apply(number: Int): TreeNode[_]
    Definition Classes
    TreeNode
  8. def argString(maxFields: Int): String
    Definition Classes
    TreeNode
  9. def asCode: String
    Definition Classes
    TreeNode
  10. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  11. final lazy val canonicalized: SparkPlan
    Definition Classes
    QueryPlan
    Annotations
    @transient()
  12. val child: SparkPlan
    Definition Classes
    GpuCustomShuffleReaderExec → UnaryExecNode
  13. final def children: Seq[SparkPlan]
    Definition Classes
    UnaryExecNode → TreeNode
  14. def childrenCoalesceGoal: Seq[CoalesceGoal]

    A goal to coalesce batches as the input to this operation.

    A goal to coalesce batches as the input to this operation. In some cases an operation will only work if all of the data is in a single batch. In other cases it may be much faster if it is in a single batch, but can tolerate multiple batches. This provides a way to express those desires.

    Definition Classes
    GpuExec
  15. def cleanupResources(): Unit
    Attributes
    protected[sql]
    Definition Classes
    SparkPlan
  16. def clone(): SparkPlan
    Definition Classes
    TreeNode → AnyRef
  17. def coalesceAfter: Boolean

    If true is returned batches after this will be coalesced.

    If true is returned batches after this will be coalesced. This should really be used in cases where it is known that the size of a batch may shrink a lot.

    Definition Classes
    GpuExec
  18. def collect[B](pf: PartialFunction[SparkPlan, B]): Seq[B]
    Definition Classes
    TreeNode
  19. def collectFirst[B](pf: PartialFunction[SparkPlan, B]): Option[B]
    Definition Classes
    TreeNode
  20. def collectLeaves(): Seq[SparkPlan]
    Definition Classes
    TreeNode
  21. def collectWithSubqueries[B](f: PartialFunction[SparkPlan, B]): Seq[B]
    Definition Classes
    QueryPlan
  22. def conf: SQLConf
    Definition Classes
    SQLConfHelper
  23. lazy val containsChild: Set[TreeNode[_]]
    Definition Classes
    TreeNode
  24. def copyTagsFrom(other: SparkPlan): Unit
    Definition Classes
    TreeNode
  25. def createAverageMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  26. def createFileCacheMetrics(): Map[String, GpuMetric]
    Attributes
    protected
    Definition Classes
    GpuExec
  27. def createMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  28. def createNanoTimingMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  29. def createSizeMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  30. def createTimingMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  31. def disableCoalesceUntilInput(): Boolean

    Returns true if there is something in the exec that cannot work when batches between multiple file partitions are combined into a single batch (coalesce).

    Returns true if there is something in the exec that cannot work when batches between multiple file partitions are combined into a single batch (coalesce).

    Definition Classes
    GpuExec
  32. def doCanonicalize(): SparkPlan

    Defines how the canonicalization should work for the current plan.

    Defines how the canonicalization should work for the current plan.

    Attributes
    protected
    Definition Classes
    GpuExec → QueryPlan
  33. def doExecute(): RDD[InternalRow]
    Attributes
    protected
    Definition Classes
    GpuCustomShuffleReaderExec → SparkPlan
  34. def doExecuteBroadcast[T](): Broadcast[T]
    Attributes
    protected[sql]
    Definition Classes
    SparkPlan
  35. final def doExecuteColumnar(): RDD[ColumnarBatch]
    Definition Classes
    GpuExec → SparkPlan
  36. def doPrepare(): Unit
    Attributes
    protected
    Definition Classes
    SparkPlan
  37. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  38. final def execute(): RDD[InternalRow]
    Definition Classes
    SparkPlan
  39. final def executeBroadcast[T](): Broadcast[T]
    Definition Classes
    SparkPlan
  40. def executeCollect(): Array[InternalRow]
    Definition Classes
    SparkPlan
  41. def executeCollectPublic(): Array[Row]
    Definition Classes
    SparkPlan
  42. final def executeColumnar(): RDD[ColumnarBatch]
    Definition Classes
    SparkPlan
  43. final def executeQuery[T](query: ⇒ T): T
    Attributes
    protected
    Definition Classes
    SparkPlan
  44. def executeTail(n: Int): Array[InternalRow]
    Definition Classes
    SparkPlan
  45. def executeTake(n: Int): Array[InternalRow]
    Definition Classes
    SparkPlan
  46. def executeToIterator(): Iterator[InternalRow]
    Definition Classes
    SparkPlan
  47. final def expressions: Seq[Expression]
    Definition Classes
    QueryPlan
  48. def fastEquals(other: TreeNode[_]): Boolean
    Definition Classes
    TreeNode
  49. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  50. def find(f: (SparkPlan) ⇒ Boolean): Option[SparkPlan]
    Definition Classes
    TreeNode
  51. def flatMap[A](f: (SparkPlan) ⇒ TraversableOnce[A]): Seq[A]
    Definition Classes
    TreeNode
  52. def foreach(f: (SparkPlan) ⇒ Unit): Unit
    Definition Classes
    TreeNode
  53. def foreachUp(f: (SparkPlan) ⇒ Unit): Unit
    Definition Classes
    TreeNode
  54. def formattedNodeName: String
    Attributes
    protected
    Definition Classes
    QueryPlan
  55. def generateTreeString(depth: Int, lastChildren: Seq[Boolean], append: (String) ⇒ Unit, verbose: Boolean, prefix: String, addSuffix: Boolean, maxFields: Int, printNodeId: Boolean, indent: Int): Unit
    Definition Classes
    TreeNode
  56. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  57. def getTagValue[T](tag: TreeNodeTag[T]): Option[T]
    Definition Classes
    TreeNode
  58. def getTaskMetrics: Option[GpuTaskMetrics]
    Definition Classes
    GpuExec
  59. def gpuExpressions: Seq[Expression]

    Return the expressions for this plan node that should be GPU expressions.

    Return the expressions for this plan node that should be GPU expressions. For most nodes this will be the same as the list of expressions, but some nodes use CPU expressions directly in some cases and will need to override this.

    Definition Classes
    GpuExec
  60. def gpuLongMetric(name: String): GpuMetric
    Definition Classes
    GpuExec
  61. def hasCoalescedPartition: Boolean
  62. def hasSkewedPartition: Boolean
  63. def hashCode(): Int
    Definition Classes
    TreeNode → AnyRef → Any
  64. val id: Int
    Definition Classes
    SparkPlan
  65. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  66. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  67. def innerChildren: Seq[QueryPlan[_]]
    Definition Classes
    QueryPlan → TreeNode
  68. def inputSet: AttributeSet
    Definition Classes
    QueryPlan
  69. def internalDoExecuteColumnar(): RDD[ColumnarBatch]

    Produces the result of the query as an RDD[ColumnarBatch] if supportsColumnar returns true.

    Produces the result of the query as an RDD[ColumnarBatch] if supportsColumnar returns true. By convention the executor that creates a ColumnarBatch is responsible for closing it when it is no longer needed. This allows input formats to be able to reuse batches if needed.

    Attributes
    protected
    Definition Classes
    GpuCustomShuffleReaderExecGpuExec
  70. def isCanonicalizedPlan: Boolean
    Attributes
    protected
    Definition Classes
    QueryPlan
  71. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  72. def isLocalReader: Boolean
  73. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  74. def jsonFields: List[JField]
    Attributes
    protected
    Definition Classes
    TreeNode
  75. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  76. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  77. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  78. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  79. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  80. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  81. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  82. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  83. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  84. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  85. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  86. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  87. def logicalLink: Option[LogicalPlan]
    Definition Classes
    SparkPlan
  88. def longMetric(name: String): SQLMetric
    Definition Classes
    SparkPlan
  89. def makeCopy(newArgs: Array[AnyRef]): SparkPlan
    Definition Classes
    SparkPlan → TreeNode
  90. def map[A](f: (SparkPlan) ⇒ A): Seq[A]
    Definition Classes
    TreeNode
  91. def mapChildren(f: (SparkPlan) ⇒ SparkPlan): SparkPlan
    Definition Classes
    TreeNode
  92. def mapExpressions(f: (Expression) ⇒ Expression): GpuCustomShuffleReaderExec.this.type
    Definition Classes
    QueryPlan
  93. def mapProductIterator[B](f: (Any) ⇒ B)(implicit arg0: ClassTag[B]): Array[B]
    Attributes
    protected
    Definition Classes
    TreeNode
  94. final lazy val metrics: Map[String, SQLMetric]
    Definition Classes
    GpuExec → SparkPlan
  95. final def missingInput: AttributeSet
    Definition Classes
    QueryPlan
  96. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  97. def nodeName: String
    Definition Classes
    TreeNode
  98. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  99. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  100. def numberedTreeString: String
    Definition Classes
    TreeNode
  101. val origin: Origin
    Definition Classes
    TreeNode
  102. def otherCopyArgs: Seq[AnyRef]
    Attributes
    protected
    Definition Classes
    TreeNode
  103. def output: Seq[Attribute]
    Definition Classes
    GpuCustomShuffleReaderExec → QueryPlan
  104. val outputBatchesLevel: MetricsLevel
    Attributes
    protected
    Definition Classes
    GpuExec
  105. def outputBatching: CoalesceGoal

    Lets a SparkPlan indicate what guarantees, if any, its output batch has.

    Lets a SparkPlan indicate what guarantees, if any, its output batch has. This lets us bypass GpuCoalesceBatch calls where ever possible. Returning a null indicates no guarantee at all, which is the default.

    Definition Classes
    GpuCustomShuffleReaderExecGpuExec
  106. def outputOrdering: Seq[SortOrder]
    Definition Classes
    SparkPlan
  107. lazy val outputPartitioning: Partitioning
    Definition Classes
    GpuCustomShuffleReaderExec → SparkPlan
  108. val outputRowsLevel: MetricsLevel
    Attributes
    protected
    Definition Classes
    GpuExec
  109. lazy val outputSet: AttributeSet
    Definition Classes
    QueryPlan
    Annotations
    @transient()
  110. def p(number: Int): SparkPlan
    Definition Classes
    TreeNode
  111. val partitionSpecs: Seq[ShufflePartitionSpec]
  112. final def prepare(): Unit
    Definition Classes
    SparkPlan
  113. def prepareSubqueries(): Unit
    Attributes
    protected
    Definition Classes
    SparkPlan
  114. def prettyJson: String
    Definition Classes
    TreeNode
  115. def printSchema(): Unit
    Definition Classes
    QueryPlan
  116. def producedAttributes: AttributeSet
    Definition Classes
    QueryPlan
  117. lazy val references: AttributeSet
    Definition Classes
    QueryPlan
    Annotations
    @transient()
  118. def requiredChildDistribution: Seq[Distribution]
    Definition Classes
    SparkPlan
  119. def requiredChildOrdering: Seq[Seq[SortOrder]]
    Definition Classes
    SparkPlan
  120. def resetMetrics(): Unit
    Definition Classes
    SparkPlan
  121. final def sameResult(other: SparkPlan): Boolean
    Definition Classes
    QueryPlan
  122. lazy val schema: StructType
    Definition Classes
    QueryPlan
  123. def schemaString: String
    Definition Classes
    QueryPlan
  124. final def semanticHash(): Int
    Definition Classes
    QueryPlan
  125. def setLogicalLink(logicalPlan: LogicalPlan): Unit
    Definition Classes
    SparkPlan
  126. def setTagValue[T](tag: TreeNodeTag[T], value: T): Unit
    Definition Classes
    TreeNode
  127. def setTaskMetrics(gpuTaskMetrics: GpuTaskMetrics): Unit
    Definition Classes
    GpuExec
  128. def simpleString(maxFields: Int): String
    Definition Classes
    QueryPlan → TreeNode
  129. def simpleStringWithNodeId(): String
    Definition Classes
    QueryPlan → TreeNode
  130. def sparkContext: SparkContext
    Attributes
    protected
    Definition Classes
    SparkPlan
  131. def sparkSession: SparkSession
    Definition Classes
    GpuExec
  132. final val sqlContext: SQLContext
    Definition Classes
    SparkPlan
  133. def statePrefix: String
    Attributes
    protected
    Definition Classes
    QueryPlan
  134. def stringArgs: Iterator[Any]
    Definition Classes
    GpuCustomShuffleReaderExec → TreeNode
  135. def subqueries: Seq[SparkPlan]
    Definition Classes
    QueryPlan
  136. def subqueriesAll: Seq[SparkPlan]
    Definition Classes
    QueryPlan
  137. def supportsColumnar: Boolean
    Definition Classes
    GpuExec → SparkPlan
  138. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  139. def toJSON: String
    Definition Classes
    TreeNode
  140. def toString(): String
    Definition Classes
    TreeNode → AnyRef → Any
  141. def transform(rule: PartialFunction[SparkPlan, SparkPlan]): SparkPlan
    Definition Classes
    TreeNode
  142. def transformAllExpressions(rule: PartialFunction[Expression, Expression]): GpuCustomShuffleReaderExec.this.type
    Definition Classes
    QueryPlan
  143. def transformDown(rule: PartialFunction[SparkPlan, SparkPlan]): SparkPlan
    Definition Classes
    TreeNode
  144. def transformExpressions(rule: PartialFunction[Expression, Expression]): GpuCustomShuffleReaderExec.this.type
    Definition Classes
    QueryPlan
  145. def transformExpressionsDown(rule: PartialFunction[Expression, Expression]): GpuCustomShuffleReaderExec.this.type
    Definition Classes
    QueryPlan
  146. def transformExpressionsUp(rule: PartialFunction[Expression, Expression]): GpuCustomShuffleReaderExec.this.type
    Definition Classes
    QueryPlan
  147. def transformUp(rule: PartialFunction[SparkPlan, SparkPlan]): SparkPlan
    Definition Classes
    TreeNode
  148. def transformUpWithNewOutput(rule: PartialFunction[SparkPlan, (SparkPlan, Seq[(Attribute, Attribute)])], skipCond: (SparkPlan) ⇒ Boolean, canGetOutput: (SparkPlan) ⇒ Boolean): SparkPlan
    Definition Classes
    QueryPlan
  149. def treeString(append: (String) ⇒ Unit, verbose: Boolean, addSuffix: Boolean, maxFields: Int, printOperatorId: Boolean): Unit
    Definition Classes
    TreeNode
  150. final def treeString(verbose: Boolean, addSuffix: Boolean, maxFields: Int, printOperatorId: Boolean): String
    Definition Classes
    TreeNode
  151. final def treeString: String
    Definition Classes
    TreeNode
  152. def unsetTagValue[T](tag: TreeNodeTag[T]): Unit
    Definition Classes
    TreeNode
  153. def vectorTypes: Option[Seq[String]]
    Definition Classes
    SparkPlan
  154. def verboseString(maxFields: Int): String
    Definition Classes
    QueryPlan → TreeNode
  155. def verboseStringWithOperatorId(): String
    Definition Classes
    UnaryExecNode → QueryPlan
  156. def verboseStringWithSuffix(maxFields: Int): String
    Definition Classes
    TreeNode
  157. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  158. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  159. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  160. def waitForSubqueries(): Unit
    Attributes
    protected
    Definition Classes
    SparkPlan
  161. def withNewChildren(newChildren: Seq[SparkPlan]): SparkPlan
    Definition Classes
    TreeNode

Inherited from GpuExec

Inherited from ShimUnaryExecNode

Inherited from UnaryExecNode

Inherited from SparkPlan

Inherited from Serializable

Inherited from Serializable

Inherited from Logging

Inherited from QueryPlan[SparkPlan]

Inherited from SQLConfHelper

Inherited from TreeNode[SparkPlan]

Inherited from Product

Inherited from Equals

Inherited from AnyRef

Inherited from Any

Ungrouped