frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove ProcessGroup from TensorPipeAgent initialization (#66708)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66708

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

Test Plan: Imported from OSS

Reviewed By: anjali411

Differential Revision: D31762735

Pulled By: H-Huang

fbshipit-source-id: 9f3879fca6b8258f7e6171b14d2c1d6cce21627d
This commit is contained in:
Howard Huang
2021-11-01 14:13:10 -07:00
committed by Facebook GitHub Bot
parent ba369ea053
commit 0cbfd466d2
11 changed files with 2 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/cpp/rpc/test_e2e_tensorpipe.cpp
View File

@ -21,10 +21,6 @@ class TestE2ETensorPipe : public TestE2EBase {
::c10d::ProcessGroupGloo::createDeviceForHostname(serverAddress));
float rpcTimeout = 30;
// Initialize server rpc agent.
auto pg = c10::make_intrusive<c10d::ProcessGroupGloo>(
store, 0, numWorkers, options);
TensorPipeRpcBackendOptions opts(
/*numWorkerThreads=*/std::max(16U, std::thread::hardware_concurrency()),
/*transports=*/nullopt,
@ -37,7 +33,6 @@ class TestE2ETensorPipe : public TestE2EBase {
"worker",
0,
numWorkers,
pg,
opts,
std::unordered_map<std::string, DeviceMap>{},
std::vector<c10::Device>{},