Issue can be closed when there is an argument annotation framework available for users to be adopted when invoking `cub::DeviceBatchedTopK`.
Issue can be closed when there is an argument annotation framework available for users to be adopted when invoking
cub::DeviceBatchedTopK.