GPU reductions improvements and bug fixes