NVIDIA / NVIDIA/cudf

[FEA] Support broadcasting aggregated expressions in groupby/rolling for cudf-polars

Open
#20,565 3 comments 0 reactions 0 assignees View on GitHub
cudf-polars feature request
Dominant language
C++
Stars
9.8k
Forks
1.1k
Avg merge
3d 6m
Merged PRs (30d)
278

Description

```python
import polars as pl

pl.Config.set_verbose(True)

lf = pl.LazyFrame(
{
"a": [1, 2, 3, 4, 5, 6],
"b": [1, 2, 3, 4, 5, 6],
"g": ["x", "x", "x", "y", "y", "y"],
}
)

print(
lf.group_by("g")
.agg((pl.col("a") - pl.col("a").mean()).dot(pl.col("b") - pl.col("b").mean()))
.collect(engine="gpu")
)

"""
/opt/miniconda3_py312/lib/python3.12/site-packages/polars/lazyframe/frame.py:2407: PerformanceWarning: Query execution with GPU not possible: unsupported operations.
The errors were:
- NotImplementedError: Broadcasting aggregated expressions in groupby/rolling
return wrap_df(ldf.collect(engine, callback))
"""
```

The polars version is '25.12.00a362'

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.