pytorch

3b194fd5 - Revert "Add offsets-based reduction to segment_reduce (CPU, CUDA)"

Revert "Add offsets-based reduction to segment_reduce (CPU, CUDA)" This reverts commit 1ec30a6647be35d123a741a39cab8b4253c1cbe0. Reverted https://github.com/pytorch/pytorch/pull/78907 on behalf of https://github.com/osalpekar due to Caused Typecasting errors in PT Distributed and fx2trt builds internally