Add XLA / TPU device type, backend type and type id (#16763)
authorAlex Şuhan <asuhan@google.com>
Tue, 5 Feb 2019 20:20:21 +0000 (12:20 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 5 Feb 2019 20:56:44 +0000 (12:56 -0800)
commit9811a4220d8949fee6915a9b7a9878dc53094eed
tree8c2323d1c5fc8933bd7791fd7011498fb07c4af4
parent6efa40e07b8e0b3ec27e4d2818172e4ed4239a90
Add XLA / TPU device type, backend type and type id (#16763)

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

Replicate the easy bits in https://github.com/pytorch/pytorch/pull/15153 with TPU / XLA instead of MSNPU. Also don't initialize the storage for XLA tensors for now.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16585

Reviewed By: ezyang

Differential Revision: D13912118

Pulled By: gchanan

fbshipit-source-id: 4889177e2478768fb281ed075b71146d1d850bd9
13 files changed:
aten/src/ATen/core/Type.h
aten/src/ATen/gen.py
aten/src/ATen/test/CMakeLists.txt
aten/src/ATen/test/xla_tensor_test.cpp [new file with mode: 0644]
aten/tools/run_tests.sh
c10/core/Backend.h
c10/core/DeviceType.cpp
c10/core/DeviceType.h
c10/core/TensorImpl.h
c10/core/TensorOptions.h
c10/core/TensorTypeIdRegistration.cpp
c10/core/TensorTypeIdRegistration.h
caffe2/proto/caffe2.proto