cleanup

Signed-off-by: Raza Jafri <rjafri@nvidia.com>
NVIDIA · Feb 22, 2021 · 3a0a098 · 3a0a098
1 parent e797eee
commit 3a0a098
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -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
@@ -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 =>