pytorch

1141b45e - Index reduction CUDA support

Index reduction CUDA support Pull Request resolved: https://github.com/pytorch/pytorch/pull/76296 Approved by: https://github.com/cpuhrsch, https://github.com/ngimel