From a94df056ae31960e59dde91c7519bd3a8e22e9d9 Mon Sep 17 00:00:00 2001 From: Haoyang Li Date: Thu, 23 Apr 2026 12:23:06 +0800 Subject: [PATCH 1/2] Re-enable accelerated columnar-to-row path The workaround from #12699 forced isAcceleratedTransposeSupported to false on every architecture to avoid the data corruption in #10062. That has now been fixed upstream in spark-rapids-jni by correcting the off-by-one in detail::determine_tiles (NVIDIA/spark-rapids-jni#4493), so restore the original post-Pascal gating. Closes #10062 --- .../com/nvidia/spark/rapids/GpuColumnarToRowExec.scala | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala index fcb71c8ef6d..215e6b2e8e3 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2025, NVIDIA CORPORATION. + * Copyright (c) 2019-2026, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -19,7 +19,7 @@ package com.nvidia.spark.rapids import scala.annotation.tailrec import scala.collection.mutable.Queue -import ai.rapids.cudf.{HostColumnVector, Table} +import ai.rapids.cudf.{Cuda, HostColumnVector, Table} import com.nvidia.spark.rapids.Arm.{closeOnExcept, withResource} import com.nvidia.spark.rapids.AssertUtils.assertInTests import com.nvidia.spark.rapids.RapidsPluginImplicits._ @@ -394,9 +394,7 @@ object GpuColumnarToRowExec { // Check if the current CUDA device architecture exceeds Pascal. // i.e. CUDA compute capability > 6.x. // Reference: https://developer.nvidia.com/cuda-gpus - //Cuda.getComputeCapabilityMajor > 6 - // https://github.com/NVIDIA/spark-rapids/issues/10062, at least until we can debug and fix it. - false + Cuda.getComputeCapabilityMajor > 6 } def makeIteratorFunc( From 8ee1db263b2591c50fa00d8f2a25a68e3fe3ef86 Mon Sep 17 00:00:00 2001 From: Haoyang Li Date: Thu, 23 Apr 2026 12:30:18 +0800 Subject: [PATCH 2/2] signoff Signed-off-by: Haoyang Li