Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Commonize v2 shim [databricks] #4235

Merged
merged 6 commits into from
Dec 3, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -405,7 +405,6 @@
<source>${project.basedir}/src/main/311+-all/scala</source>
<source>${project.basedir}/src/main/311until320-noncdh/scala</source>
<source>${project.basedir}/src/main/31xdb/scala</source>
<source>${project.basedir}/src/main/312db/scala</source>
jlowe marked this conversation as resolved.
Show resolved Hide resolved
<source>${project.basedir}/src/main/post320-treenode/scala</source>
</sources>
</configuration>
Expand Down Expand Up @@ -445,12 +444,12 @@
<source>${project.basedir}/src/main/301+-nondb/scala</source>
<source>${project.basedir}/src/main/301until320-all/scala</source>
<source>${project.basedir}/src/main/301until320-nondb/scala</source>
<source>${project.basedir}/src/main/301until330-all/scala</source>
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👍

<source>${project.basedir}/src/main/311+-all/scala</source>
<source>${project.basedir}/src/main/311+-nondb/scala</source>
<source>${project.basedir}/src/main/311until320-all/scala</source>
<source>${project.basedir}/src/main/311until320-noncdh/scala</source>
<source>${project.basedir}/src/main/311until320-nondb/scala</source>
<source>${project.basedir}/src/main/301until330-all/scala</source>
<source>${project.basedir}/src/main/pre320-treenode/scala</source>
</sources>
</configuration>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark301Shims extends SparkBaseShims
with Spark30Xuntil33XShims {
class Spark301Shims extends Spark30XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ package com.nvidia.spark.rapids.shims.spark301db
import com.nvidia.spark.rapids._
import com.nvidia.spark.rapids.shims.v2._

class Spark301dbShims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark301dbShims extends Spark30XdbShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION
}
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark302Shims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark302Shims extends Spark30XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark303Shims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark303Shims extends Spark30XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark304Shims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark304Shims extends Spark30XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark311Shims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark311Shims extends Spark31XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark311CDHShims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark311CDHShims extends Spark31XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark312Shims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark312Shims extends Spark31XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark312dbShims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark312dbShims extends Spark31XdbShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ import org.apache.parquet.schema.MessageType
import org.apache.spark.sql.execution.datasources.parquet.ParquetFilters
import org.apache.spark.sql.internal.SQLConf

class Spark313Shims extends SparkBaseShims with Spark30Xuntil33XShims {
class Spark313Shims extends Spark31XShims with Spark30Xuntil33XShims {

override def getSparkShimVersion: ShimVersion = SparkShimServiceProvider.VERSION

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ import org.apache.spark.sql.types._
import org.apache.spark.storage.{BlockId, BlockManagerId}
import org.apache.spark.unsafe.types.CalendarInterval

abstract class SparkBaseShims extends Spark30XShims with Logging {
abstract class Spark30XdbShims extends Spark30XdbShimsBase with Logging {
override def getParquetFilters(
schema: MessageType,
pushDownDate: Boolean,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ import org.apache.spark.sql.rapids.execution.GpuCustomShuffleReaderExec
/**
* Shim base class that can be compiled with every supported 3.0.x
*/
trait Spark30XShims extends SparkShims {
trait Spark30XdbShimsBase extends SparkShims {
override def parquetRebaseReadKey: String =
SQLConf.LEGACY_PARQUET_REBASE_MODE_IN_READ.key
override def parquetRebaseWriteKey: String =
Expand Down
Loading