Skip to content

Commit c867097

Browse files
Add missing CUDA stream to cudf-polars left-semi join (#20398)
#20291 missed a spot in `Join` where we need to pass the CUDA stream to the pylibcudf join function. This shows up in PDSH query 4. Authors: - Tom Augspurger (https://github.com/TomAugspurger) Approvers: - Matthew Murray (https://github.com/Matt711) URL: #20398
1 parent e568adf commit c867097

File tree

1 file changed

+1
-1
lines changed
  • python/cudf_polars/cudf_polars/dsl

1 file changed

+1
-1
lines changed

python/cudf_polars/cudf_polars/dsl/ir.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2359,7 +2359,7 @@ def do_evaluate(
23592359
join_fn, left_policy, right_policy = cls._joiners(how)
23602360
if right_policy is None:
23612361
# Semi join
2362-
lg = join_fn(left_on.table, right_on.table, null_equality)
2362+
lg = join_fn(left_on.table, right_on.table, null_equality, stream)
23632363
table = plc.copying.gather(left.table, lg, left_policy, stream=stream)
23642364
result = DataFrame.from_table(
23652365
table, left.column_names, left.dtypes, stream=stream

0 commit comments

Comments
 (0)