pytorch

9291bb9d - [13/N] Update gather with CPU/CUDA implementations

[13/N] Update gather with CPU/CUDA implementations ghstack-source-id: af1d9ccdc9dca0080e808b400bbc8096b3a818db Pull Request resolved: https://github.com/pytorch/pytorch/pull/86409