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(