pytorch/benchmarks/distributed
Garrett Cramer 304c02ee44 refactor ps benchmark (#60784)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60784

This pr refactors the ps benchmark for modular trainers.

Test Plan: Imported from OSS

Reviewed By: zou3519

Differential Revision: D29697291

Pulled By: gcramer23

fbshipit-source-id: 64579a1f5326d3cd9f32936dcf53bc243d54b71d
2021-07-14 13:19:13 -07:00
..
ddp Fix benchmarks/distributed/ddp/benchmark.py (#51095) 2021-01-29 11:10:13 -08:00
pipeline faster generate_square_subsequent_mask in nn.Transformer (#60631) 2021-06-25 16:07:01 -07:00
rpc refactor ps benchmark (#60784) 2021-07-14 13:19:13 -07:00