fix arange shape issue inconsistency across cpu and cuda (#18462)
authorSoumith Chintala <soumith@gmail.com>
Tue, 26 Mar 2019 22:23:43 +0000 (15:23 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 26 Mar 2019 22:27:24 +0000 (15:27 -0700)
commit436723122e646b1100a07f04a59043a58a8bd1f0
tree32929c3860400032a33a0d079cb3feb3f2d53864
parentbbe110f4e1ed28da3fa528071a1985453ef36caa
fix arange shape issue inconsistency across cpu and cuda (#18462)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/18363
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18462

Differential Revision: D14620263

Pulled By: soumith

fbshipit-source-id: 223524cdda2f5d55c2ca8d4cdcf6f7a05a6c15eb
aten/src/ATen/native/RangeFactories.cpp
aten/src/ATen/native/cuda/RangeFactories.cu
test/test_torch.py