Interesting, I guess we implemented this op properly :)

It is strange that cub does not guarantee determinism here - it's not too hard an op to implement in CUDA.