[VL] Enable Spark repeat function (apache#6419)

yma11 · Jul 12, 2024 · 62d55cf · 62d55cf
1 parent e95f32a
commit 62d55cf
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/backends-velox/src/test/scala/org/apache/gluten/execution/ScalarFunctionsValidateSuite.scala b/backends-velox/src/test/scala/org/apache/gluten/execution/ScalarFunctionsValidateSuite.scala
@@ -1270,4 +1270,10 @@ class ScalarFunctionsValidateSuite extends FunctionsValidateTest {
       }
     }
   }
+
+  test("repeat") {
+    runQueryAndCompare("select repeat(c_comment, 5) from customer limit 50") {
+      checkGlutenOperatorMatch[ProjectExecTransformer]
+    }
+  }
 }
diff --git a/cpp/velox/substrait/SubstraitToVeloxPlanValidator.cc b/cpp/velox/substrait/SubstraitToVeloxPlanValidator.cc
@@ -65,7 +65,6 @@ static const std::unordered_set<std::string> kBlackList = {
     "concat_ws",
     "from_json",
     "json_array_length",
-    "repeat",
     "trunc",
     "sequence",
     "approx_percentile",

diff --git a/docs/velox-backend-support-progress.md b/docs/velox-backend-support-progress.md
@@ -171,7 +171,7 @@ Gluten supports 199 functions. (Drag to right to see all data types)
 | parse_url                     |                        |                       |        |                          |         |      |       |     |      |       |        |      |           |        |         |      |        |          |       |     |        |     |
 | position                      | strpos                 |                       |        |                          |         |      |       |     |      |       |        |      |           |        |         |      |        |          |       |     |        |     |
 | printf                        |                        |                       |        |                          |         |      |       |     |      |       |        |      |           |        |         |      |        |          |       |     |        |     |
-| repeat                        |                        |                       |        |                          |         |      |       |     |      |       |        |      |           |        |         |      |        |          |       |     |        |     |
+| repeat                        |                        | repeat                      | S       |                          |         |      |       |     |      |       |        |      |           |        |         |      |        |          |       |     |        |     |
 | replace                       | replace                | replace               | S      |                          |         |      |       |     |      |       |        |      |           | S      |         |      |        |          |       |     |        |     |
 | reverse                       | reverse                |                       | S      |                          |         |      |       |     |      |       |        |      |           | S      |         |      |        |          |       |     |        |     |
 | right                         |                        |                       |        |                          |         |      |       |     |      |       |        |      |           |        |         |      |        |          |       |     |        |     |