Right now, we support CUB based reduction, but some routines such as max, min, sum, or scan use some CUB special functions.

We should add a note to the docstring of such functions stating that cub acceleration is available.