projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 8d5b950) | patch
Replace std::unordered_map<c10::Device, c10::Device> with DeviceMap (#64393)
authorPavel Belevich <pbelevich@fb.com>
Thu, 2 Sep 2021 07:57:39 +0000 (00:57 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 2 Sep 2021 08:36:19 +0000 (01:36 -0700)
commitee8a6c1d141ae49e23323bdd485fb4b390541f69
treeef08348ed42945f972d408c9aaa195c65f41f8actree | snapshot
parent8d5b95019d69d43963b33a1b188ad1fec8079664commit | diff
Replace std::unordered_map<c10::Device, c10::Device> with DeviceMap (#64393)

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

cc pietern mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse agolynski SciPioneer H-Huang mrzzd cbalioglu gcramer23

Test Plan: Imported from OSS

Reviewed By: rohan-varma

Differential Revision: D30708384

Pulled By: pbelevich

fbshipit-source-id: 1c565727e4f09cd9e560874dd90aa403470b4a97
torch/csrc/distributed/autograd/functions/recvrpc_backward.cpp diff | blob | history
torch/csrc/distributed/autograd/functions/recvrpc_backward.h diff | blob | history
torch/csrc/distributed/autograd/rpc_messages/rpc_with_autograd.cpp diff | blob | history
torch/csrc/distributed/autograd/rpc_messages/rpc_with_autograd.h diff | blob | history
torch/csrc/distributed/autograd/utils.cpp diff | blob | history
torch/csrc/distributed/autograd/utils.h diff | blob | history
torch/csrc/distributed/rpc/request_callback_no_python.cpp diff | blob | history
torch/csrc/distributed/rpc/rpc_agent.h diff | blob | history
torch/csrc/distributed/rpc/testing/faulty_tensorpipe_agent.cpp diff | blob | history
torch/csrc/distributed/rpc/testing/faulty_tensorpipe_agent.h diff | blob | history
torch/csrc/distributed/rpc/utils.cpp diff | blob | history
Domain: Machine Learning / ML Framework;