From 3a0a09824f4aa186b1b9823218beec97c51dd37e Mon Sep 17 00:00:00 2001 From: Raza Jafri Date: Sun, 21 Feb 2021 16:55:01 -0800 Subject: [PATCH] cleanup Signed-off-by: Raza Jafri --- .../main/scala/com/nvidia/spark/rapids/GpuParquetScan.scala | 2 +- .../main/scala/org/apache/spark/sql/rapids/arithmetic.scala | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuParquetScan.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuParquetScan.scala index acc2516b6f72..1da744831b69 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuParquetScan.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuParquetScan.scala @@ -747,7 +747,7 @@ abstract class FileParquetPartitionReaderBase( if (cols.nonEmpty) { // create a new struct column with the new ones withResource(cols) { newCols => - ColumnVector.replaceColumnsInStruct(cv, newColIndices.result(), newCols) + cv.replaceColumnsInStruct(newColIndices.result(), newCols) } } else { cv diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/arithmetic.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/arithmetic.scala index 7859cb7355ea..048eb02e7f07 100644 --- a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/arithmetic.scala +++ b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/arithmetic.scala @@ -354,8 +354,8 @@ case class GpuDivide(left: Expression, right: Expression) extends GpuDivModLike val decimalType = createCudfDecimal(10, Math.max(l.scale, r.scale)) val outputType = dataType.asInstanceOf[DecimalType] val cudfOutputType = createCudfDecimal(outputType.precision, outputType.scale) - withResource(lhs.getBase().castDecimal32ToDecimal64(decimalType)) { decimalLhs => - withResource(rhs.getBase.castDecimal32ToDecimal64(decimalType)) { decimalRhs => + withResource(lhs.getBase().castTo(decimalType)) { decimalLhs => + withResource(rhs.getBase.castTo(decimalType)) { decimalRhs => val tmp = decimalLhs.div(decimalRhs, implicitReturnType(decimalLhs, decimalRhs)) if (tmp.getType != cudfOutputType) { withResource(tmp) { tmp =>