[Perf branch] Add support for 2d reduction and bug fix#2629

jataylo merged 1 commit intoiupaikov_perf_tuning_rocm71ROCm/pytorch:iupaikov_perf_tuning_rocm71from

jataylo-patch-4ROCm/pytorch:jataylo-patch-4Copy head branch name to clipboard