Packages

o

com.nvidia.spark.rapids.shims

SparkShimImpl

object SparkShimImpl extends Spark320PlusShims with Spark320PlusNonDBShims with Spark31Xuntil33XShims with AnsiCastRuleShims

Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. SparkShimImpl
  2. AnsiCastRuleShims
  3. Spark31Xuntil33XShims
  4. Spark320PlusNonDBShims
  5. Spark320PlusShims
  6. Logging
  7. RebaseShims
  8. SparkShims
  9. AnyRef
  10. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

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

    dropped by SPARK-34234

    dropped by SPARK-34234

    Definition Classes
    Spark320PlusShims → SparkShims
  12. final def avroRebaseReadKey: String
    Definition Classes
    RebaseShims
  13. final def avroRebaseWriteKey: String
    Definition Classes
    RebaseShims
  14. final def broadcastModeTransform(mode: BroadcastMode, rows: Array[InternalRow]): Any
    Definition Classes
    Spark320PlusNonDBShims → SparkShims
  15. def checkCToRWithExecBroadcastAQECoalPart(p: SparkPlan, parent: Option[SparkPlan]): Boolean
    Definition Classes
    SparkShims
  16. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  17. def columnarAdaptivePlan(a: AdaptiveSparkPlanExec, goal: CoalesceSizeGoal): SparkPlan
    Definition Classes
    Spark320PlusShims → SparkShims
  18. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  19. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  20. final def filesFromFileIndex(fileIndex: PartitioningAwareFileIndex): Seq[FileStatus]
    Definition Classes
    Spark320PlusNonDBShims → SparkShims
  21. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  22. 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
  23. def getAdaptiveInputPlan(adaptivePlan: AdaptiveSparkPlanExec): SparkPlan
    Definition Classes
    Spark320PlusShims → SparkShims
  24. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  25. def getDataWriteCmds: Map[Class[_ <: DataWritingCommand], DataWritingCommandRule[_ <: DataWritingCommand]]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  26. def getDateFormatter(): DateFormatter
    Definition Classes
    Spark320PlusShims → SparkShims
  27. def getExecs: Map[Class[_ <: SparkPlan], ExecRule[_ <: SparkPlan]]
    Definition Classes
    Spark320PlusShims → SparkShims
  28. def getExprs: Map[Class[_ <: Expression], ExprRule[_ <: Expression]]
    Definition Classes
    Spark320PlusShims → SparkShims
  29. def getFileScanRDD(sparkSession: SparkSession, readFunction: (PartitionedFile) ⇒ Iterator[InternalRow], filePartitions: Seq[FilePartition], readDataSchema: StructType, metadataColumns: Seq[AttributeReference], fileFormat: Option[FileFormat]): RDD[InternalRow]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  30. def getParquetFilters(schema: MessageType, pushDownDate: Boolean, pushDownTimestamp: Boolean, pushDownDecimal: Boolean, pushDownStartWith: Boolean, pushDownInFilterThreshold: Int, caseSensitive: Boolean, lookupFileMeta: (String) ⇒ String, dateTimeRebaseModeFromConf: String): ParquetFilters
    Definition Classes
    SparkShimImpl → SparkShims
  31. def getRunnableCmds: Map[Class[_ <: RunnableCommand], RunnableCommandRule[_ <: RunnableCommand]]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  32. def getScans: Map[Class[_ <: Scan], ScanRule[_ <: Scan]]
    Definition Classes
    Spark320PlusShims → SparkShims
  33. def getShuffleFromCToRWithExecBroadcastAQECoalPart(p: SparkPlan): Option[SparkPlan]
    Definition Classes
    SparkShims
  34. def getWindowExpressions(winPy: WindowInPandasExec): Seq[NamedExpression]
    Definition Classes
    Spark320PlusNonDBShims
  35. def hasAliasQuoteFix: Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  36. def hasCastFloatTimestampUpcast: Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  37. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  38. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  39. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  40. def int96ParquetRebaseRead(conf: SQLConf): String
    Definition Classes
    RebaseShims
  41. def int96ParquetRebaseReadKey: String
    Definition Classes
    RebaseShims
  42. def int96ParquetRebaseWrite(conf: SQLConf): String
    Definition Classes
    RebaseShims
  43. def int96ParquetRebaseWriteKey: String
    Definition Classes
    RebaseShims
  44. final def isAqePlan(p: SparkPlan): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  45. def isCastingStringToNegDecimalScaleSupported: Boolean
    Definition Classes
    SparkShimImpl → SparkShims
  46. def isCustomReaderExec(x: SparkPlan): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  47. def isEmptyRelation(relation: Any): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  48. final def isExchangeOp(plan: SparkPlanMeta[_]): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  49. def isExecutorBroadcastShuffle(shuffle: ShuffleExchangeLike): Boolean
    Definition Classes
    SparkShims
  50. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  51. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  52. def isWindowFunctionExec(plan: SparkPlan): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  53. def leafNodeDefaultParallelism(ss: SparkSession): Int
    Definition Classes
    Spark320PlusShims → SparkShims
  54. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  55. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  56. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  57. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  58. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  59. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  60. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  61. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  62. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  63. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  64. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  65. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  66. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  67. def neverReplaceShowCurrentNamespaceCommand: ExecRule[_ <: SparkPlan]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  68. final def newBroadcastQueryStageExec(old: BroadcastQueryStageExec, newPlan: SparkPlan): BroadcastQueryStageExec
    Definition Classes
    Spark320PlusNonDBShims → SparkShims
  69. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  70. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  71. final def parquetRebaseRead(conf: SQLConf): String
    Definition Classes
    RebaseShims
  72. final def parquetRebaseReadKey: String
    Definition Classes
    RebaseShims
  73. final def parquetRebaseWrite(conf: SQLConf): String
    Definition Classes
    RebaseShims
  74. final def parquetRebaseWriteKey: String
    Definition Classes
    RebaseShims
  75. 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
    SparkShimImpl → Spark320PlusShims → SparkShims
  76. def reusedExchangeExecPfn: PartialFunction[SparkPlan, ReusedExchangeExec]

    Case class ShuffleQueryStageExec holds an additional field shuffleOrigin affecting the unapply method signature

    Case class ShuffleQueryStageExec holds an additional field shuffleOrigin affecting the unapply method signature

    Definition Classes
    Spark320PlusNonDBShims → SparkShims
  77. final def sessionFromPlan(plan: SparkPlan): SparkSession
    Definition Classes
    Spark320PlusShims → SparkShims
  78. def shouldFailDivOverflow(): Boolean
    Definition Classes
    Spark320PlusShims → SparkShims
  79. def shuffleParentReadsShuffleData(shuffle: ShuffleExchangeLike, parent: SparkPlan): Boolean
    Definition Classes
    SparkShims
  80. 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
  81. 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
  82. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  83. def toString(): String
    Definition Classes
    AnyRef → Any
  84. 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
  85. def v1RepairTableCommand(tableName: TableIdentifier): RunnableCommand
    Definition Classes
    Spark320PlusShims → SparkShims
  86. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  87. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  88. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from AnsiCastRuleShims

Inherited from Spark31Xuntil33XShims

Inherited from Spark320PlusNonDBShims

Inherited from Spark320PlusShims

Inherited from Logging

Inherited from RebaseShims

Inherited from SparkShims

Inherited from AnyRef

Inherited from Any

Ungrouped