Fix math::Set for large tensor (#17539)
authorXiaomeng Yang <yangxm@fb.com>
Wed, 27 Feb 2019 20:18:52 +0000 (12:18 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 27 Feb 2019 20:34:58 +0000 (12:34 -0800)
commit9709d5e787616a4dde1f0d120db951d103d1d857
treee77be5516fce21ba9fe52a0b6c53ae188843eee5
parentb4572668b48b746802580263dafa12762d202379
Fix math::Set for large tensor (#17539)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17539

Fix math::Set for large tensor

i-am-not-moving-c2-to-c10

Reviewed By: dzhulgakov, houseroad

Differential Revision: D14240756

fbshipit-source-id: 0ade26790be41fb26d2cc193bfa3082c7bd4e69d
caffe2/contrib/aten/aten_op.cc
caffe2/utils/math/elementwise.cc
caffe2/utils/math/elementwise.cu
caffe2/utils/math/elementwise.h