Fixes reduction launch config (#64304)
authorXiang Gao <qasdfgtyuiop@gmail.com>
Wed, 1 Sep 2021 17:17:52 +0000 (10:17 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 1 Sep 2021 17:30:40 +0000 (10:30 -0700)
commitc4f3f6e62d3201852fc9aac8f736d528a5a36bfe
treeddb2ea7c1c200cfa407f113d69ccf2bac7340515
parentd5bfdd3dac33dfa84e2a511fa79c4ad4e0e6b822
Fixes reduction launch config (#64304)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/48573
See also https://github.com/pytorch/pytorch/pull/64194

Pull Request resolved: https://github.com/pytorch/pytorch/pull/64304

Reviewed By: janeyx99

Differential Revision: D30689600

Pulled By: ngimel

fbshipit-source-id: bf2103ca177fd3b6e27bc0324b81925234483a29
aten/src/ATen/native/cuda/LinearAlgebra.cu
aten/src/ATen/native/cuda/Normalization.cu
aten/src/ATen/native/cuda/Reduce.cuh