pytorch
0e9a295b - Refactor GlooDeviceFactory::makeDeviceFor... (#58996)

Commit
3 years ago
Refactor GlooDeviceFactory::makeDeviceFor... (#58996) Summary: `makeDeviceForHostname` and `makeDeviceForInterface` are almost duplicate except for different default argument values Create generic `makeGlooDevice` anonymous function that takes both host name and interface name and call it from both makeDeviceFor[Hostname|Interface] Also solve two other minor issues: - do not call `getenv("GLOO_DEVICE_TRANSPORT")` during library load time - Raise exception rather than crash if GLOO_DEVICE_TRANSPORT is set to unknown value Fixes #{issue number} Pull Request resolved: https://github.com/pytorch/pytorch/pull/58996 Reviewed By: pbelevich Differential Revision: D28713324 Pulled By: malfet fbshipit-source-id: cb33b438078d163e3ec6f047f2e5247b07d94f8d
Author
Parents
Loading