Back out "Revert D13596031: Improve c2-aten tensor interop and add proper testing...
authorDmytro Dzhulgakov <dzhulgakov@fb.com>
Thu, 31 Jan 2019 21:30:58 +0000 (13:30 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 31 Jan 2019 21:38:55 +0000 (13:38 -0800)
commita061e3fd77d2e3b11ba90b81e1e3e4a0c35c7403
tree6a7e65eeec3ca8566dab963776da65520b93e372
parent0b29bd82f6a9ecada2daa43e755ac72f46146da5
Back out "Revert D13596031: Improve c2-aten tensor interop and add proper testing" (#16514)

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

Original commit changeset: dc371697f14b
Relanding https://github.com/pytorch/pytorch/pull/15860 - the problem was that layer_norm was using at::empty which is not yet on mobile

Reviewed By: ezyang

Differential Revision: D13861480

fbshipit-source-id: e2116da32bc117175c96b9151b1beba9b31eff36
12 files changed:
aten/src/ATen/core/LegacyTypeDispatch.h
aten/src/ATen/core/Tensor.cpp
aten/src/ATen/core/Tensor.h
aten/src/ATen/templates/Tensor.h
aten/src/ATen/test/CMakeLists.txt
aten/src/ATen/test/cuda_tensor_interop_test.cpp [new file with mode: 0644]
aten/src/ATen/test/tensor_interop_test.cpp
aten/src/THC/generic/THCTensorMathPointwise.cu
aten/tools/run_tests.sh
caffe2/core/tensor.cc
caffe2/core/tensor.h
caffe2/operators/layer_norm_op.cc