Optimize CPU version performance of the nonzero function. (#15925)
authorVitaly Fedyunin <vitalyf@fb.com>
Fri, 11 Jan 2019 01:47:49 +0000 (17:47 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 11 Jan 2019 01:50:38 +0000 (17:50 -0800)
commit0934e8de58bffcc89b80dd136580188512a476c0
treed1cbb0ea69e850a2d6ec73dee809e107ff4af772
parent890568a018b9043878547cf22465d23270731a6f
Optimize CPU version performance of the nonzero function. (#15925)

Summary:
Same as #15190 but compatible with MSVS compiler
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15925

Differential Revision: D13623473

Pulled By: VitalyFedyunin

fbshipit-source-id: d0db9dbc1a0d8fc9bda08348cb1d3763ae9f8679
aten/src/TH/generic/THTensorEvenMoreMath.cpp