projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 01b8162) | patch
Compute cuda reduction buffer size in elements (#63969)
authorNatalia Gimelshein <ngimel@fb.com>
Thu, 26 Aug 2021 01:17:10 +0000 (18:17 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 26 Aug 2021 01:18:37 +0000 (18:18 -0700)
commitf4bc28990f6edcaf6bfc9e9737a70ea0be514198
tree23ab327e4fd4ce1da723e32a14c82d3b4b171cc2tree | snapshot
parent01b8162d00bfb0844a3f8a165d49907e51a16addcommit | diff
Compute cuda reduction buffer size in elements (#63969)

Summary:
Resubmit of https://github.com/pytorch/pytorch/issues/63885

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

Reviewed By: mruberry

Differential Revision: D30549423

Pulled By: ngimel

fbshipit-source-id: b16d25030d44ced789c125a333d72b02a8f45067
aten/src/ATen/native/cuda/Reduce.cuh diff | blob | history
test/test_reductions.py diff | blob | history
Domain: Machine Learning / ML Framework;