diff --git a/docs/supported_ops.md b/docs/supported_ops.md index 511a3611803..9824842169b 100644 --- a/docs/supported_ops.md +++ b/docs/supported_ops.md @@ -143,7 +143,7 @@ Accelerator supports are described below. S S* S -NS +S* NS NS NS diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOverrides.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOverrides.scala index a111a8e6993..1f7ebc214fc 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOverrides.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOverrides.scala @@ -2396,7 +2396,7 @@ object GpuOverrides { }), exec[CollectLimitExec]( "Reduce to single partition and apply limit", - ExecChecks(TypeSig.commonCudfTypes, TypeSig.all), + ExecChecks(TypeSig.commonCudfTypes + TypeSig.DECIMAL, TypeSig.all), (collectLimitExec, conf, p, r) => new GpuCollectLimitMeta(collectLimitExec, conf, p, r)) .disabledByDefault("Collect Limit replacement can be slower on the GPU, if huge number " + "of rows in a batch it could help by limiting the number of rows transferred from " +