pytorch/test/cpp
Luca Wehrstedt 0422e67336 Use Devices instead of DeviceIndexes in TensorPipe agent (#57294)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57294

With the advent of CPUs in the device maps, and to be more generic (e.g., to support AMD GPUs), and to avoid conversions when passing to Future and RRef and such, it's easier to use Devices instead of DeviceIndices. This started by just migrating the TensorPipe agent but the RPC layer is quite intertwined so I had to migrate a lot of stuff.
ghstack-source-id: 127916562

Test Plan: CI

Reviewed By: mrshenli

Differential Revision: D28092733

fbshipit-source-id: 024dcb3648c5898ab13e770413c43958f04f1a8a
2021-05-01 16:12:55 -07:00
..
api Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
common
dist_autograd Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
jit [AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT 2021-04-29 04:12:25 -07:00
lite_interpreter_runtime Revert D27958477: [PyTorch][Edge] Add v4 and v5 models and remove unused model 2021-04-23 14:42:01 -07:00
rpc Use Devices instead of DeviceIndexes in TensorPipe agent (#57294) 2021-05-01 16:12:55 -07:00
tensorexpr [pytorch][nnc] update external functions for mobile build (#56850) 2021-04-30 19:07:19 -07:00
__init__.py