Merge pull request #9910 from NVIDIA/branch-23.12

[auto-merge] branch-23.12 to branch-24.02 [skip ci] [bot]
NVIDIA · Nov 30, 2023 · fcad227 · fcad227
2 parents b3f1e47 + c389ae0
commit fcad227
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/integration_tests/src/main/python/ast_test.py b/integration_tests/src/main/python/ast_test.py
@@ -380,3 +380,9 @@ def test_or(data_gen):
                        f.col('a') | f.lit(True),
                        f.lit(False) | f.col('b'),
                        f.col('a') | f.col('b')))
+
+def test_multi_tier_ast():
+    assert_gpu_ast(
+        is_supported=True,
+        func=lambda spark: spark.range(10).withColumn("x", f.col("id")).repartition(1)\
+            .selectExpr("(id < x) == (id < (id + x))"))