saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/cpp/rpc
History
Lucas Hosseini 8868f9c8e3 [TensorPipe] Use targetDevice in tensorpipe_agent. (#56346) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56346

Now that TensorPipe's API has `targetDevice`, use that instead of
manually writing the CUDA device index in `metadata`.

Test Plan: CI

Reviewed By: lw

Differential Revision: D27703235

fbshipit-source-id: c5b620e3b3ce619367412efdbe9fa3778f6b8869
2021-04-20 11:54:13 -07:00
..
CMakeLists.txt fix tensorpipe_agent linked even when USE_TENSORPIPE is turned off (#53281) 2021-03-04 13:29:27 -08:00
e2e_test_base.cpp
e2e_test_base.h [RPC] Refactor rref_context to not use utils::Future (#51697) 2021-02-19 00:59:38 -08:00
test_e2e_process_group.cpp [distributed] add base processgroup::options (#53662) 2021-03-17 18:40:04 -07:00
test_e2e_tensorpipe.cpp [distributed] add base processgroup::options (#53662) 2021-03-17 18:40:04 -07:00
test_tensorpipe_serialization.cpp [TensorPipe] Use targetDevice in tensorpipe_agent. (#56346) 2021-04-20 11:54:13 -07:00
test_wire_serialization.cpp