NVIDIA / NVIDIA/cudf

[FEA] Support `sort` in groupby/rolling

Open
#20,546 4 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)
print(
pl.LazyFrame(
{
"value": [3, 2, 1],
"group": ['a', 'a', 'b']
}
)
.select(pl.col('value').sort().over('group'))
.collect(engine=pl.GPUEngine(executor='in-memory'))
)

"""
/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: No support for in groupby/rolling
return wrap_df(ldf.collect(engine, callback))
shape: (3, 1)
┌───────┐
│ value │
│ --- │
│ i64 │
╞═══════╡
│ 2 │
│ 3 │
│ 1 │
└───────┘
"""
```

The cudf-polars version is '25.12.00a350'

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.