Packages

t

com.nvidia.spark.rapids.shims

Spark320PlusShims

trait Spark320PlusShims extends SparkShims with RebaseShims with Logging

Shim base class that can be compiled with every supported 3.2.0+

Linear Supertypes
Logging, RebaseShims, SparkShims, AnyRef, Any
Known Subclasses
SparkShimImpl
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. Spark320PlusShims
  2. Logging
  3. RebaseShims
  4. SparkShims
  5. AnyRef
  6. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def ansiCastRule: ExprRule[_ <: Expression]

    Return the replacement rule for AnsiCast.

    Return the replacement rule for AnsiCast. 'AnsiCast' is removed from Spark 3.4.0, so need to handle it separately.

    Definition Classes
    SparkShims
  2. abstract def broadcastModeTransform(mode: BroadcastMode, toArray: Array[InternalRow]): Any
    Definition Classes
    SparkShims
  3. abstract def filesFromFileIndex(fileCatalog: PartitioningAwareFileIndex): Seq[FileStatus]
    Definition Classes
    SparkShims
  4. abstract def getDataWriteCmds: Map[Class[_ <: DataWritingCommand], DataWritingCommandRule[_ <: DataWritingCommand]]
    Definition Classes
    SparkShims
  5. abstract def getFileScanRDD(sparkSession: SparkSession, readFunction: (PartitionedFile) ⇒ Iterator[InternalRow], filePartitions: Seq[FilePartition], readDataSchema: StructType, metadataColumns: Seq[AttributeReference] = Seq.empty, fileFormat: Option[FileFormat] = None): RDD[InternalRow]
    Definition Classes
    SparkShims
  6. abstract def getParquetFilters(schema: MessageType, pushDownDate: Boolean, pushDownTimestamp: Boolean, pushDownDecimal: Boolean, pushDownStartWith: Boolean, pushDownInFilterThreshold: Int, caseSensitive: Boolean, lookupFileMeta: (String) ⇒ String, dateTimeRebaseModeFromConf: String): ParquetFilters
    Definition Classes
    SparkShims
  7. abstract def getRunnableCmds: Map[Class[_ <: RunnableCommand], RunnableCommandRule[_ <: RunnableCommand]]
    Definition Classes
    SparkShims
  8. abstract def getWindowExpressions(winPy: WindowInPandasExec): Seq[NamedExpression]
  9. abstract def neverReplaceShowCurrentNamespaceCommand: ExecRule[_ <: SparkPlan]
    Definition Classes
    SparkShims
  10. abstract def newBroadcastQueryStageExec(old: BroadcastQueryStageExec, newPlan: SparkPlan): BroadcastQueryStageExec
    Definition Classes
    SparkShims
  11. abstract def reusedExchangeExecPfn: PartialFunction[SparkPlan, ReusedExchangeExec]
    Definition Classes
    SparkShims

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. def addExecBroadcastShuffle(p: SparkPlan): SparkPlan

    If the shim doesn't support executor broadcast, just return the plan passed in

    If the shim doesn't support executor broadcast, just return the plan passed in

    Definition Classes
    SparkShims
  5. def addRowShuffleToQueryStageTransitionIfNeeded(c2r: ColumnarToRowTransition, sqse: ShuffleQueryStageExec): SparkPlan

    Adds a row-based shuffle to the transititonal shuffle query stage if needed.

    Adds a row-based shuffle to the transititonal shuffle query stage if needed. This is needed when AQE plans a GPU shuffleexchange to be reused by a parent plan exec that consumes rows

    Definition Classes
    SparkShims
  6. def applyPostShimPlanRules(plan: SparkPlan): SparkPlan
    Definition Classes
    SparkShims
  7. def applyShimPlanRules(plan: SparkPlan, conf: RapidsConf): SparkPlan
    Definition Classes
    SparkShims
  8. final def aqeShuffleReaderExec: ExecRule[_ <: SparkPlan]
    Definition Classes
    Spark320PlusShims → SparkShims
  9. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  10. def attachTreeIfSupported[TreeType <: TreeNode[_], A](tree: TreeType, msg: String)(f: ⇒ A): A

    dropped by SPARK-34234

    dropped by SPARK-34234

    Definition Classes
    Spark320PlusShims → SparkShims
  11. final def avroRebaseReadKey: String
    Definition Classes
    RebaseShims
  12. final def avroRebaseWriteKey: String
    Definition Classes
    RebaseShims
  13. def checkCToRWithExecBroadcastAQECoalPart(p: SparkPlan, parent: Option[SparkPlan]): Boolean
    Definition Classes
    SparkShims
  14. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  15. def columnarAdaptivePlan(a: AdaptiveSparkPlanExec, goal: CoalesceSizeGoal): SparkPlan
    Definition Classes
    Spark320PlusShims → SparkShims
  16. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  17. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  18. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  19. def findOperators(plan: SparkPlan, predicate: (SparkPlan) ⇒ Boolean): Seq[SparkPlan]

    Walk the plan recursively and return a list of operators that match the predicate

    Walk the plan recursively and return a list of operators that match the predicate

    Definition Classes
    Spark320PlusShims → SparkShims
  20. def getAdaptiveInputPlan(adaptivePlan: AdaptiveSparkPlanExec): SparkPlan
    Definition Classes
    Spark320PlusShims → SparkShims
  21. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  22. def getDateFormatter(): DateFormatter
    Definition Classes
    Spark320PlusShims → SparkShims
  23. def getExecs: Map[Class[_ <: SparkPlan], ExecRule[_ <: SparkPlan]]
    Definition Classes
    Spark320PlusShims → SparkShims
  24. def getExprs: Map[Class[_ <: Expression], ExprRule[_ <: Expression]]
    Definition Classes
    Spark320PlusShims → SparkShims
  25. def getScans: Map[Class[_ <: Scan], ScanRule[_ <: Scan]]
    Definition Classes
    Spark320PlusShims → SparkShims
  26. def getShuffleFromCToRWithExecBroadcastAQECoalPart(p: SparkPlan): Option[SparkPlan]
    Definition Classes
    SparkShims
  27. def hasAliasQuoteFix: Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  28. def hasCastFloatTimestampUpcast: Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  29. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  30. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  31. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  32. def int96ParquetRebaseRead(conf: SQLConf): String
    Definition Classes
    RebaseShims
  33. def int96ParquetRebaseReadKey: String
    Definition Classes
    RebaseShims
  34. def int96ParquetRebaseWrite(conf: SQLConf): String
    Definition Classes
    RebaseShims
  35. def int96ParquetRebaseWriteKey: String
    Definition Classes
    RebaseShims
  36. final def isAqePlan(p: SparkPlan): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  37. def isCastingStringToNegDecimalScaleSupported: Boolean
    Definition Classes
    SparkShims
  38. def isCustomReaderExec(x: SparkPlan): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  39. def isEmptyRelation(relation: Any): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  40. final def isExchangeOp(plan: SparkPlanMeta[_]): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  41. def isExecutorBroadcastShuffle(shuffle: ShuffleExchangeLike): Boolean
    Definition Classes
    SparkShims
  42. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  43. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  44. def isWindowFunctionExec(plan: SparkPlan): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  45. def leafNodeDefaultParallelism(ss: SparkSession): Int
    Definition Classes
    Spark320PlusShims → SparkShims
  46. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  47. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  48. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  49. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  50. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  51. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  52. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  53. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  54. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  55. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  56. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  57. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  58. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  59. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  60. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  61. final def parquetRebaseRead(conf: SQLConf): String
    Definition Classes
    RebaseShims
  62. final def parquetRebaseReadKey: String
    Definition Classes
    RebaseShims
  63. final def parquetRebaseWrite(conf: SQLConf): String
    Definition Classes
    RebaseShims
  64. final def parquetRebaseWriteKey: String
    Definition Classes
    RebaseShims
  65. def reproduceEmptyStringBug: Boolean

    Handle regexp_replace inconsistency from https://issues.apache.org/jira/browse/SPARK-39107

    Handle regexp_replace inconsistency from https://issues.apache.org/jira/browse/SPARK-39107

    Definition Classes
    Spark320PlusShims → SparkShims
  66. final def sessionFromPlan(plan: SparkPlan): SparkSession
    Definition Classes
    Spark320PlusShims → SparkShims
  67. def shouldFailDivOverflow(): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  68. def shuffleParentReadsShuffleData(shuffle: ShuffleExchangeLike, parent: SparkPlan): Boolean
    Definition Classes
    SparkShims
  69. def skipAssertIsOnTheGpu(plan: SparkPlan): Boolean

    Our tests, by default, will check that all operators are running on the GPU, but there are some operators that we do not translate to GPU plans, so we need a way to bypass the check for those.

    Our tests, by default, will check that all operators are running on the GPU, but there are some operators that we do not translate to GPU plans, so we need a way to bypass the check for those.

    Definition Classes
    Spark320PlusShims → SparkShims
  70. def supportsColumnarAdaptivePlans: Boolean

    Determine if the Spark version allows the supportsColumnar flag to be overridden in AdaptiveSparkPlanExec.

    Determine if the Spark version allows the supportsColumnar flag to be overridden in AdaptiveSparkPlanExec. This feature was introduced in Spark 3.2 as part of SPARK-35881.

    Definition Classes
    Spark320PlusShims → SparkShims
  71. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  72. def toString(): String
    Definition Classes
    AnyRef → Any
  73. def tryTransformIfEmptyRelation(mode: BroadcastMode): Option[Any]

    This call can produce an EmptyHashedRelation or an empty array, allowing the AQE rule EliminateJoinToEmptyRelation in Spark 3.1.x to optimize certain joins.

    This call can produce an EmptyHashedRelation or an empty array, allowing the AQE rule EliminateJoinToEmptyRelation in Spark 3.1.x to optimize certain joins.

    In Spark 3.2.0, the optimization is still performed (under AQEPropagateEmptyRelation), but the AQE optimizer is looking at the metrics for the query stage to determine if numRows == 0, and if so it can eliminate certain joins.

    The call is implemented only for Spark 3.1.x+. It is disabled in Databricks because it requires a task context to perform the BroadcastMode.transform call, but we'd like to call this from the driver.

    Definition Classes
    Spark320PlusShims → SparkShims
  74. def v1RepairTableCommand(tableName: TableIdentifier): RunnableCommand
    Definition Classes
    Spark320PlusShims → SparkShims
  75. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  76. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  77. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from Logging

Inherited from RebaseShims

Inherited from SparkShims

Inherited from AnyRef

Inherited from Any

Ungrouped