Packages

c

com.nvidia.spark.rapids

GpuHashAggregateExec

case class GpuHashAggregateExec(requiredChildDistributionExpressions: Option[Seq[Expression]], groupingExpressions: Seq[NamedExpression], aggregateExpressions: Seq[GpuAggregateExpression], aggregateAttributes: Seq[Attribute], resultExpressions: Seq[NamedExpression], child: SparkPlan, configuredTargetBatchSize: Long, configuredTieredProjectEnabled: Boolean, estimatedPreProcessGrowth: Double, forceSinglePassAgg: Boolean, allowSinglePassAgg: Boolean) extends SparkPlan with ShimUnaryExecNode with GpuExec with Product with Serializable

The GPU version of SortAggregateExec that is intended for partial aggregations that are not reductions and so it sorts the input data ahead of time to do it in a single pass.

requiredChildDistributionExpressions

this is unchanged by the GPU. It is used in EnsureRequirements to be able to add shuffle nodes

groupingExpressions

The expressions that, when applied to the input batch, return the grouping key

aggregateExpressions

The GpuAggregateExpression instances for this node

aggregateAttributes

References to each GpuAggregateExpression (attribute references)

resultExpressions

the expected output expression of this hash aggregate (which this node should project)

child

incoming plan (where we get input columns from)

configuredTargetBatchSize

user-configured maximum device memory size of a batch

configuredTieredProjectEnabled

configurable optimization to use tiered projections

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. GpuHashAggregateExec
  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 GpuHashAggregateExec(requiredChildDistributionExpressions: Option[Seq[Expression]], groupingExpressions: Seq[NamedExpression], aggregateExpressions: Seq[GpuAggregateExpression], aggregateAttributes: Seq[Attribute], resultExpressions: Seq[NamedExpression], child: SparkPlan, configuredTargetBatchSize: Long, configuredTieredProjectEnabled: Boolean, estimatedPreProcessGrowth: Double, forceSinglePassAgg: Boolean, allowSinglePassAgg: Boolean)

    requiredChildDistributionExpressions

    this is unchanged by the GPU. It is used in EnsureRequirements to be able to add shuffle nodes

    groupingExpressions

    The expressions that, when applied to the input batch, return the grouping key

    aggregateExpressions

    The GpuAggregateExpression instances for this node

    aggregateAttributes

    References to each GpuAggregateExpression (attribute references)

    resultExpressions

    the expected output expression of this hash aggregate (which this node should project)

    child

    incoming plan (where we get input columns from)

    configuredTargetBatchSize

    user-configured maximum device memory size of a batch

    configuredTieredProjectEnabled

    configurable optimization to use tiered projections

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
    GpuHashAggregateExecGpuExec
  5. val aggregateAttributes: Seq[Attribute]
  6. val aggregateExpressions: Seq[GpuAggregateExpression]
  7. lazy val allAttributes: AttributeSeq

    All the attributes that are used for this plan.

    All the attributes that are used for this plan. NOT used for aggregation

    Definition Classes
    GpuHashAggregateExec → QueryPlan
  8. lazy val allMetrics: Map[String, GpuMetric]
    Definition Classes
    GpuExec
  9. val allowSinglePassAgg: Boolean
  10. def apply(number: Int): TreeNode[_]
    Definition Classes
    TreeNode
  11. def argString(maxFields: Int): String
    Definition Classes
    TreeNode
  12. def asCode: String
    Definition Classes
    TreeNode
  13. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  14. final lazy val canonicalized: SparkPlan
    Definition Classes
    QueryPlan
    Annotations
    @transient()
  15. val child: SparkPlan
    Definition Classes
    GpuHashAggregateExec → UnaryExecNode
  16. final def children: Seq[SparkPlan]
    Definition Classes
    UnaryExecNode → TreeNode
  17. 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
  18. def cleanupResources(): Unit
    Attributes
    protected[sql]
    Definition Classes
    SparkPlan
  19. def clone(): SparkPlan
    Definition Classes
    TreeNode → AnyRef
  20. 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
  21. def collect[B](pf: PartialFunction[SparkPlan, B]): Seq[B]
    Definition Classes
    TreeNode
  22. def collectFirst[B](pf: PartialFunction[SparkPlan, B]): Option[B]
    Definition Classes
    TreeNode
  23. def collectLeaves(): Seq[SparkPlan]
    Definition Classes
    TreeNode
  24. def collectWithSubqueries[B](f: PartialFunction[SparkPlan, B]): Seq[B]
    Definition Classes
    QueryPlan
  25. def conf: SQLConf
    Definition Classes
    SQLConfHelper
  26. val configuredTargetBatchSize: Long
  27. val configuredTieredProjectEnabled: Boolean
  28. lazy val containsChild: Set[TreeNode[_]]
    Definition Classes
    TreeNode
  29. def copyTagsFrom(other: SparkPlan): Unit
    Definition Classes
    TreeNode
  30. def createAverageMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  31. def createFileCacheMetrics(): Map[String, GpuMetric]
    Attributes
    protected
    Definition Classes
    GpuExec
  32. def createMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  33. def createNanoTimingMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  34. def createSizeMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  35. def createTimingMetric(level: MetricsLevel, name: String): GpuMetric
    Attributes
    protected
    Definition Classes
    GpuExec
  36. 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
  37. 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
  38. def doExecute(): RDD[InternalRow]
    Definition Classes
    GpuHashAggregateExec → SparkPlan
  39. def doExecuteBroadcast[T](): Broadcast[T]
    Attributes
    protected[sql]
    Definition Classes
    SparkPlan
  40. final def doExecuteColumnar(): RDD[ColumnarBatch]
    Definition Classes
    GpuExec → SparkPlan
  41. def doPrepare(): Unit
    Attributes
    protected
    Definition Classes
    SparkPlan
  42. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  43. val estimatedPreProcessGrowth: Double
  44. final def execute(): RDD[InternalRow]
    Definition Classes
    SparkPlan
  45. final def executeBroadcast[T](): Broadcast[T]
    Definition Classes
    SparkPlan
  46. def executeCollect(): Array[InternalRow]
    Definition Classes
    SparkPlan
  47. def executeCollectPublic(): Array[Row]
    Definition Classes
    SparkPlan
  48. final def executeColumnar(): RDD[ColumnarBatch]
    Definition Classes
    SparkPlan
  49. final def executeQuery[T](query: ⇒ T): T
    Attributes
    protected
    Definition Classes
    SparkPlan
  50. def executeTail(n: Int): Array[InternalRow]
    Definition Classes
    SparkPlan
  51. def executeTake(n: Int): Array[InternalRow]
    Definition Classes
    SparkPlan
  52. def executeToIterator(): Iterator[InternalRow]
    Definition Classes
    SparkPlan
  53. def expectedChildOrderingIfNeeded: Seq[SortOrder]
  54. final def expressions: Seq[Expression]
    Definition Classes
    QueryPlan
  55. def fastEquals(other: TreeNode[_]): Boolean
    Definition Classes
    TreeNode
  56. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  57. def find(f: (SparkPlan) ⇒ Boolean): Option[SparkPlan]
    Definition Classes
    TreeNode
  58. def flatMap[A](f: (SparkPlan) ⇒ TraversableOnce[A]): Seq[A]
    Definition Classes
    TreeNode
  59. val forceSinglePassAgg: Boolean
  60. def foreach(f: (SparkPlan) ⇒ Unit): Unit
    Definition Classes
    TreeNode
  61. def foreachUp(f: (SparkPlan) ⇒ Unit): Unit
    Definition Classes
    TreeNode
  62. def formattedNodeName: String
    Attributes
    protected
    Definition Classes
    QueryPlan
  63. 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
  64. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  65. def getTagValue[T](tag: TreeNodeTag[T]): Option[T]
    Definition Classes
    TreeNode
  66. def getTaskMetrics: Option[GpuTaskMetrics]
    Definition Classes
    GpuExec
  67. 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
    GpuHashAggregateExecGpuExec
  68. def gpuLongMetric(name: String): GpuMetric
    Definition Classes
    GpuExec
  69. val groupingExpressions: Seq[NamedExpression]
  70. def hasAlias: Boolean
    Attributes
    protected
  71. def hashCode(): Int
    Definition Classes
    TreeNode → AnyRef → Any
  72. val id: Int
    Definition Classes
    SparkPlan
  73. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  74. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  75. def innerChildren: Seq[QueryPlan[_]]
    Definition Classes
    QueryPlan → TreeNode
  76. def inputAttributes: Seq[Attribute]
  77. def inputSet: AttributeSet
    Definition Classes
    QueryPlan
  78. def internalDoExecuteColumnar(): RDD[ColumnarBatch]
    Definition Classes
    GpuHashAggregateExecGpuExec
  79. def isCanonicalizedPlan: Boolean
    Attributes
    protected
    Definition Classes
    QueryPlan
  80. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  81. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  82. def jsonFields: List[JField]
    Attributes
    protected
    Definition Classes
    TreeNode
  83. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  84. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  85. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  86. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  87. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  88. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  89. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  90. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  91. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  92. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  93. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  94. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  95. def logicalLink: Option[LogicalPlan]
    Definition Classes
    SparkPlan
  96. def longMetric(name: String): SQLMetric
    Definition Classes
    SparkPlan
  97. def makeCopy(newArgs: Array[AnyRef]): SparkPlan
    Definition Classes
    SparkPlan → TreeNode
  98. def map[A](f: (SparkPlan) ⇒ A): Seq[A]
    Definition Classes
    TreeNode
  99. def mapChildren(f: (SparkPlan) ⇒ SparkPlan): SparkPlan
    Definition Classes
    TreeNode
  100. def mapExpressions(f: (Expression) ⇒ Expression): GpuHashAggregateExec.this.type
    Definition Classes
    QueryPlan
  101. def mapProductIterator[B](f: (Any) ⇒ B)(implicit arg0: ClassTag[B]): Array[B]
    Attributes
    protected
    Definition Classes
    TreeNode
  102. final lazy val metrics: Map[String, SQLMetric]
    Definition Classes
    GpuExec → SparkPlan
  103. final def missingInput: AttributeSet
    Definition Classes
    QueryPlan
  104. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  105. def nodeName: String
    Definition Classes
    TreeNode
  106. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  107. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  108. def numberedTreeString: String
    Definition Classes
    TreeNode
  109. val origin: Origin
    Definition Classes
    TreeNode
  110. def otherCopyArgs: Seq[AnyRef]
    Attributes
    protected
    Definition Classes
    TreeNode
  111. def output: Seq[Attribute]
    Definition Classes
    GpuHashAggregateExec → QueryPlan
  112. val outputBatchesLevel: MetricsLevel
    Attributes
    protected
    Definition Classes
    GpuHashAggregateExecGpuExec
  113. 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
    GpuExec
  114. def outputExpressions: Seq[NamedExpression]
    Attributes
    protected
  115. def outputOrdering: Seq[SortOrder]
    Definition Classes
    SparkPlan
  116. final def outputPartitioning: Partitioning
    Definition Classes
    GpuHashAggregateExec → SparkPlan
  117. val outputRowsLevel: MetricsLevel
    Attributes
    protected
    Definition Classes
    GpuHashAggregateExecGpuExec
  118. lazy val outputSet: AttributeSet
    Definition Classes
    QueryPlan
    Annotations
    @transient()
  119. def p(number: Int): SparkPlan
    Definition Classes
    TreeNode
  120. final def prepare(): Unit
    Definition Classes
    SparkPlan
  121. def prepareSubqueries(): Unit
    Attributes
    protected
    Definition Classes
    SparkPlan
  122. def prettyJson: String
    Definition Classes
    TreeNode
  123. def printSchema(): Unit
    Definition Classes
    QueryPlan
  124. def producedAttributes: AttributeSet
    Definition Classes
    GpuHashAggregateExec → QueryPlan
  125. lazy val references: AttributeSet
    Definition Classes
    QueryPlan
    Annotations
    @transient()
  126. def replaceAlias(attr: AttributeReference): Option[Attribute]
    Attributes
    protected
  127. def replaceAliases(exprs: Seq[Expression]): Seq[Expression]
    Attributes
    protected
  128. def requiredChildDistribution: List[Distribution]
    Definition Classes
    GpuHashAggregateExec → SparkPlan
  129. val requiredChildDistributionExpressions: Option[Seq[Expression]]
  130. def requiredChildOrdering: Seq[Seq[SortOrder]]
    Definition Classes
    SparkPlan
  131. def resetMetrics(): Unit
    Definition Classes
    SparkPlan
  132. val resultExpressions: Seq[NamedExpression]
  133. final def sameResult(other: SparkPlan): Boolean
    Definition Classes
    QueryPlan
  134. lazy val schema: StructType
    Definition Classes
    QueryPlan
  135. def schemaString: String
    Definition Classes
    QueryPlan
  136. final def semanticHash(): Int
    Definition Classes
    QueryPlan
  137. def setLogicalLink(logicalPlan: LogicalPlan): Unit
    Definition Classes
    SparkPlan
  138. def setTagValue[T](tag: TreeNodeTag[T], value: T): Unit
    Definition Classes
    TreeNode
  139. def setTaskMetrics(gpuTaskMetrics: GpuTaskMetrics): Unit
    Definition Classes
    GpuExec
  140. def simpleString(maxFields: Int): String
    Definition Classes
    GpuHashAggregateExec → QueryPlan → TreeNode
  141. def simpleStringWithNodeId(): String
    Definition Classes
    QueryPlan → TreeNode
  142. def sparkContext: SparkContext
    Attributes
    protected
    Definition Classes
    SparkPlan
  143. def sparkSession: SparkSession
    Definition Classes
    GpuExec
  144. final val sqlContext: SQLContext
    Definition Classes
    SparkPlan
  145. def statePrefix: String
    Attributes
    protected
    Definition Classes
    QueryPlan
  146. def stringArgs: Iterator[Any]
    Attributes
    protected
    Definition Classes
    TreeNode
  147. def subqueries: Seq[SparkPlan]
    Definition Classes
    QueryPlan
  148. def subqueriesAll: Seq[SparkPlan]
    Definition Classes
    QueryPlan
  149. def supportsColumnar: Boolean
    Definition Classes
    GpuExec → SparkPlan
  150. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  151. def toJSON: String
    Definition Classes
    TreeNode
  152. def toString(): String
    Definition Classes
    TreeNode → AnyRef → Any
  153. def transform(rule: PartialFunction[SparkPlan, SparkPlan]): SparkPlan
    Definition Classes
    TreeNode
  154. def transformAllExpressions(rule: PartialFunction[Expression, Expression]): GpuHashAggregateExec.this.type
    Definition Classes
    QueryPlan
  155. def transformDown(rule: PartialFunction[SparkPlan, SparkPlan]): SparkPlan
    Definition Classes
    TreeNode
  156. def transformExpressions(rule: PartialFunction[Expression, Expression]): GpuHashAggregateExec.this.type
    Definition Classes
    QueryPlan
  157. def transformExpressionsDown(rule: PartialFunction[Expression, Expression]): GpuHashAggregateExec.this.type
    Definition Classes
    QueryPlan
  158. def transformExpressionsUp(rule: PartialFunction[Expression, Expression]): GpuHashAggregateExec.this.type
    Definition Classes
    QueryPlan
  159. def transformUp(rule: PartialFunction[SparkPlan, SparkPlan]): SparkPlan
    Definition Classes
    TreeNode
  160. def transformUpWithNewOutput(rule: PartialFunction[SparkPlan, (SparkPlan, Seq[(Attribute, Attribute)])], skipCond: (SparkPlan) ⇒ Boolean, canGetOutput: (SparkPlan) ⇒ Boolean): SparkPlan
    Definition Classes
    QueryPlan
  161. def treeString(append: (String) ⇒ Unit, verbose: Boolean, addSuffix: Boolean, maxFields: Int, printOperatorId: Boolean): Unit
    Definition Classes
    TreeNode
  162. final def treeString(verbose: Boolean, addSuffix: Boolean, maxFields: Int, printOperatorId: Boolean): String
    Definition Classes
    TreeNode
  163. final def treeString: String
    Definition Classes
    TreeNode
  164. lazy val uniqueModes: Seq[AggregateMode]
    Attributes
    protected
  165. def unsetTagValue[T](tag: TreeNodeTag[T]): Unit
    Definition Classes
    TreeNode
  166. def vectorTypes: Option[Seq[String]]
    Definition Classes
    SparkPlan
  167. def verboseString(maxFields: Int): String
    Definition Classes
    GpuHashAggregateExec → QueryPlan → TreeNode
  168. def verboseStringWithOperatorId(): String
    Definition Classes
    GpuHashAggregateExec → UnaryExecNode → QueryPlan
  169. def verboseStringWithSuffix(maxFields: Int): String
    Definition Classes
    TreeNode
  170. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  171. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  172. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  173. def waitForSubqueries(): Unit
    Attributes
    protected
    Definition Classes
    SparkPlan
  174. 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