package execution
- Alphabetic
- Public
- All
Type Members
- abstract class BaseHashJoinIterator extends SplittableJoinIterator
-
class
BatchPartitionIdPassthrough extends Partitioner
A dummy partitioner for use with records whose partition ids have been pre-computed (i.e.
A dummy partitioner for use with records whose partition ids have been pre-computed (i.e. for use on RDDs of (Int, Row) pairs where the Int is a partition id in the expected range).
-
class
CoalescedBatchPartitioner extends Partitioner
A Partitioner that might group together one or more partitions from the parent.
-
class
ConditionalHashJoinIterator extends BaseHashJoinIterator
An iterator that does a hash join against a stream of batches with an inequality condition.
An iterator that does a hash join against a stream of batches with an inequality condition. The compiled condition will be closed when this iterator is closed.
- class ConditionalNestedLoopExistenceJoinIterator extends ExistenceJoinIterator
- class ConditionalNestedLoopJoinIterator extends SplittableJoinIterator
-
class
CrossJoinIterator extends AbstractGpuJoinIterator
An iterator that does a cross join against a stream of batches.
-
abstract
class
ExistenceJoinIterator extends Iterator[ColumnarBatch] with TaskAutoCloseableResource with Arm
Existence join generates an
existsboolean column withtrueorfalsein it, then appends it to theoutputcolumns.Existence join generates an
existsboolean column withtrueorfalsein it, then appends it to theoutputcolumns. The true inexistscolumn indicates left table should retain that row, the row number ofexistsequals to the row number of left table.e.g.:
select * from left_table where left_table.column_0 >= 3 or exists (select * from right_table where left_table.column_1 < right_table.column_1) Explanation of this sql is: Filter(left_table.column_0 >= 3 or `exists`) Existence_join (left + `exists`) // `exists` do not shrink or expand the rows of left table left_table right_table - case class GpuBroadcastExchangeExec(mode: BroadcastMode, child: SparkPlan) extends GpuBroadcastExchangeExecBase with Product with Serializable
- abstract class GpuBroadcastExchangeExecBase extends Exchange with ShimBroadcastExchangeLike with ShimUnaryExecNode with GpuExec
- class GpuBroadcastMeta extends SparkPlanMeta[BroadcastExchangeExec]
- case class GpuBroadcastNestedLoopJoinExec(left: SparkPlan, right: SparkPlan, joinType: JoinType, gpuBuildSide: GpuBuildSide, condition: Option[Expression], targetSizeBytes: Long) extends SparkPlan with ShimBinaryExecNode with GpuExec with Product with Serializable
- class GpuBroadcastNestedLoopJoinMeta extends GpuBroadcastJoinMeta[BroadcastNestedLoopJoinExec]
- class GpuColumnToRowMapPartitionsRDD extends MapPartitionsRDD[InternalRow, ColumnarBatch]
-
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.
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.
- trait GpuHashJoin extends SparkPlan with GpuExec
-
abstract
class
GpuShuffleExchangeExecBase extends Exchange with ShimUnaryExecNode with GpuExec
Performs a shuffle that will result in the desired partitioning.
-
abstract
class
GpuShuffleExchangeExecBaseWithMetrics extends GpuShuffleExchangeExecBase
Performs a shuffle that will result in the desired partitioning.
- class GpuShuffleMeta extends SparkPlanMeta[ShuffleExchangeExec]
- case class GpuSubqueryBroadcastExec(name: String, index: Int, buildKeys: Seq[Expression], child: SparkPlan)(modeKeys: Option[Seq[Expression]]) extends BaseSubqueryExec with GpuExec with ShimUnaryExecNode with Product with Serializable
- class GpuSubqueryBroadcastMeta extends SparkPlanMeta[SubqueryBroadcastExec]
-
class
HashJoinIterator extends BaseHashJoinIterator
An iterator that does a hash join against a stream of batches.
- class HashedExistenceJoinIterator extends ExistenceJoinIterator
-
class
SerializeBatchDeserializeHostBuffer extends Serializable with AutoCloseable with Arm
- Annotations
- @SerialVersionUID()
-
class
SerializeConcatHostBuffersDeserializeBatch extends Serializable with Arm with AutoCloseable
- Annotations
- @SerialVersionUID()
-
class
ShuffledBatchRDD extends RDD[ColumnarBatch]
This is a specialized version of
org.apache.spark.rdd.ShuffledRDDthat is optimized for shufflingColumnarBatchinstead of Java key-value pairs.This is a specialized version of
org.apache.spark.rdd.ShuffledRDDthat is optimized for shufflingColumnarBatchinstead of Java key-value pairs.This RDD takes a
ShuffleDependency(dependency), and an array ofShufflePartitionSpecas input arguments.The
dependencyhas the parent RDD of this RDD, which represents the dataset before shuffle (i.e. map output). Elements of this RDD are (partitionId, Row) pairs. Partition ids should be in the range [0, numPartitions - 1].dependency.partitioneris the original partitioner used to partition map output, anddependency.partitioner.numPartitionsis the number of pre-shuffle partitions (i.e. the number of partitions of the map output).This code is made to try and match the Spark code as closely as possible to make maintenance simpler. Fixing compiler or IDE warnings in this code may not be ideal if the same warnings are in Spark.
- case class ShuffledBatchRDDPartition(index: Int, spec: ShufflePartitionSpec) extends Partition with Product with Serializable
Value Members
- object GpuBroadcastExchangeExecBase extends Serializable
- object GpuBroadcastHelper
- object GpuBroadcastNestedLoopJoinExec extends Arm with Serializable
- object GpuColumnToRowMapPartitionsRDD extends Serializable
- object GpuHashJoin extends Arm with Serializable
- object GpuShuffleExchangeExecBase extends Serializable
- object GpuShuffleMeta
- object GpuSubqueryBroadcastExec extends Serializable
-
object
InternalColumnarRddConverter extends Logging
Please don't use this class directly use com.nvidia.spark.rapids.ColumnarRdd instead.
Please don't use this class directly use com.nvidia.spark.rapids.ColumnarRdd instead. We had to place the implementation in a spark specific package to poke at the internals of spark more than anyone should know about.
This provides a way to get back out GPU Columnar data RDD[Table]. Each Table will have the same schema as the dataframe passed in. If the schema of the dataframe is something that Rapids does not currently support an
IllegalArgumentExceptionwill be thrown.The size of each table will be determined by what is producing that table but typically will be about the number of bytes set by
RapidsConf.GPU_BATCH_SIZE_BYTES.Table is not a typical thing in an RDD so special care needs to be taken when working with it. By default it is not serializable so repartitioning the RDD or any other operator that involves a shuffle will not work. This is because it is very expensive to serialize and deserialize a GPU Table using a conventional spark shuffle. Also most of the memory associated with the Table is on the GPU itself, so each table must be closed when it is no longer needed to avoid running out of GPU memory. By convention it is the responsibility of the one consuming the data to close it when they no longer need it.
- object JoinTypeChecks
- object SerializedHostTableUtils extends Arm
- object ShimTrampolineUtil
- object TrampolineUtil