rapidsai / node

GPU-accelerated data science and visualization in node
https://rapidsai.github.io/node/
Apache License 2.0
187 stars 20 forks source link

Using CUB from node-rapids #410

Closed ghalimi closed 2 years ago

ghalimi commented 2 years ago

Is there a way to use CUB from node-rapids? More specifically, is there a way to invoke OneSweep?

trxcllnt commented 2 years ago

We don't have any explicit bindings into cub, but DataFrame and Series sortValues() methods are implemented via Thrust (and Thrust uses cub::DeviceRadixSort under the hood).

ghalimi commented 2 years ago

Awesome! Thank you so much.