Packages

o

com.nvidia.spark.rapids.shims

SparkShimImpl

object SparkShimImpl extends Spark31XShims

Linear Supertypes
Spark31XShims, Logging, Spark31Xuntil33XShims, SparkShims, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. SparkShimImpl
  2. Spark31XShims
  3. Logging
  4. Spark31Xuntil33XShims
  5. SparkShims
  6. AnyRef
  7. 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 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
  5. 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
    Spark31XShims → SparkShims
  6. def applyPostShimPlanRules(plan: SparkPlan): SparkPlan
    Definition Classes
    SparkShims
  7. def applyShimPlanRules(plan: SparkPlan, conf: RapidsConf): SparkPlan
    Definition Classes
    SparkShims
  8. def aqeShuffleReaderExec: ExecRule[_ <: SparkPlan]
    Definition Classes
    Spark31XShims → 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
    Spark31XShims → SparkShims
  11. def avroRebaseReadKey: String
    Definition Classes
    Spark31XShims → SparkShims
  12. def avroRebaseWriteKey: String
    Definition Classes
    Spark31XShims → SparkShims
  13. def broadcastModeTransform(mode: BroadcastMode, rows: Array[InternalRow]): Any
    Definition Classes
    Spark31XShims → SparkShims
  14. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  15. def columnarAdaptivePlan(a: AdaptiveSparkPlanExec, goal: CoalesceSizeGoal): SparkPlan
    Definition Classes
    Spark31XShims → SparkShims
  16. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  17. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  18. def filesFromFileIndex(fileIndex: PartitioningAwareFileIndex): Seq[FileStatus]
    Definition Classes
    Spark31XShims → SparkShims
  19. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  20. 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
    Spark31XShims → SparkShims
  21. def getAdaptiveInputPlan(adaptivePlan: AdaptiveSparkPlanExec): SparkPlan
    Definition Classes
    Spark31XShims → SparkShims
  22. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  23. def getDataWriteCmds: Map[Class[_ <: DataWritingCommand], DataWritingCommandRule[_ <: DataWritingCommand]]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  24. def getDateFormatter(): DateFormatter
    Definition Classes
    Spark31XShims → SparkShims
  25. def getExecs: Map[Class[_ <: SparkPlan], ExecRule[_ <: SparkPlan]]
    Definition Classes
    Spark31XShims → SparkShims
  26. def getExprs: Map[Class[_ <: Expression], ExprRule[_ <: Expression]]
    Definition Classes
    Spark31XShims → SparkShims
  27. def getFileScanRDD(sparkSession: SparkSession, readFunction: (PartitionedFile) ⇒ Iterator[InternalRow], filePartitions: Seq[FilePartition], readDataSchema: StructType, metadataColumns: Seq[AttributeReference]): RDD[InternalRow]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  28. 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
  29. def getRunnableCmds: Map[Class[_ <: RunnableCommand], RunnableCommandRule[_ <: RunnableCommand]]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  30. def getScans: Map[Class[_ <: Scan], ScanRule[_ <: Scan]]
    Definition Classes
    Spark31XShims → SparkShims
  31. def hasAliasQuoteFix: Boolean
    Definition Classes
    Spark31XShims → SparkShims
  32. def hasCastFloatTimestampUpcast: Boolean
    Definition Classes
    SparkShimImpl → Spark31XShims → SparkShims
  33. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  34. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  35. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  36. def int96ParquetRebaseRead(conf: SQLConf): String
    Definition Classes
    Spark31XShims → SparkShims
  37. def int96ParquetRebaseReadKey: String
    Definition Classes
    Spark31XShims → SparkShims
  38. def int96ParquetRebaseWrite(conf: SQLConf): String
    Definition Classes
    Spark31XShims → SparkShims
  39. def int96ParquetRebaseWriteKey: String
    Definition Classes
    Spark31XShims → SparkShims
  40. def isAqePlan(p: SparkPlan): Boolean
    Definition Classes
    Spark31XShims → SparkShims
  41. def isCastingStringToNegDecimalScaleSupported: Boolean
    Definition Classes
    Spark31XShims → SparkShims
  42. def isCustomReaderExec(x: SparkPlan): Boolean
    Definition Classes
    Spark31XShims → SparkShims
  43. def isEmptyRelation(relation: Any): Boolean
    Definition Classes
    Spark31XShims → SparkShims
  44. def isExchangeOp(plan: SparkPlanMeta[_]): Boolean
    Definition Classes
    Spark31XShims → SparkShims
  45. def isExecutorBroadcastShuffle(shuffle: ShuffleExchangeLike): Boolean
    Definition Classes
    SparkShims
  46. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  47. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  48. def isWindowFunctionExec(plan: SparkPlan): Boolean
    Definition Classes
    Spark31XShims → SparkShims
  49. def leafNodeDefaultParallelism(ss: SparkSession): Int
    Definition Classes
    Spark31XShims → SparkShims
  50. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  51. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  52. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  53. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  54. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  55. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  56. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  57. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  58. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  59. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  60. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  61. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  62. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  63. def neverReplaceShowCurrentNamespaceCommand: ExecRule[_ <: SparkPlan]
    Definition Classes
    Spark31Xuntil33XShims → SparkShims
  64. def newBroadcastQueryStageExec(old: BroadcastQueryStageExec, newPlan: SparkPlan): BroadcastQueryStageExec
    Definition Classes
    Spark31XShims → SparkShims
  65. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  66. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  67. def parquetRebaseRead(conf: SQLConf): String
    Definition Classes
    Spark31XShims → SparkShims
  68. def parquetRebaseReadKey: String
    Definition Classes
    Spark31XShims → SparkShims
  69. def parquetRebaseWrite(conf: SQLConf): String
    Definition Classes
    Spark31XShims → SparkShims
  70. def parquetRebaseWriteKey: String
    Definition Classes
    Spark31XShims → SparkShims
  71. 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 → Spark31XShims → SparkShims
  72. def reusedExchangeExecPfn: PartialFunction[SparkPlan, ReusedExchangeExec]
    Definition Classes
    Spark31XShims → SparkShims
  73. def sessionFromPlan(plan: SparkPlan): SparkSession
    Definition Classes
    Spark31XShims → SparkShims
  74. def shouldFailDivOverflow: Boolean
    Definition Classes
    Spark31XShims → SparkShims
  75. def shuffleParentReadsShuffleData(shuffle: ShuffleExchangeLike, parent: SparkPlan): Boolean
    Definition Classes
    SparkShims
  76. 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
    Spark31XShims → SparkShims
  77. 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
    Spark31XShims → SparkShims
  78. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  79. def toString(): String
    Definition Classes
    AnyRef → Any
  80. 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
    Spark31XShims → SparkShims
  81. def v1RepairTableCommand(tableName: TableIdentifier): RunnableCommand
    Definition Classes
    Spark31XShims → SparkShims
  82. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  83. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  84. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from Spark31XShims

Inherited from Logging

Inherited from Spark31Xuntil33XShims

Inherited from SparkShims

Inherited from AnyRef

Inherited from Any

Ungrouped