Xuehai Pan
c0ed38e644
[BE][Easy][3/19] enforce style for empty lines in import segments in benchmarks/ ( #129754 )
...
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501 . Most changes are auto-generated by linter.
You can review these PRs via:
```bash
git diff --ignore-all-space --ignore-blank-lines HEAD~1
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129754
Approved by: https://github.com/ezyang
2024-07-17 14:34:42 +00:00
Edward Z. Yang
dd3a77bc96
Apply UFMT to all files in benchmarks/ ( #105928 )
...
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105928
Approved by: https://github.com/albanD
2023-07-26 01:18:48 +00:00
Sean Lawlor
34c9f5a8da
[DDP Communication Hook] Update get_tensor and set_tensor to be cleaner naming conventions (buffer() and set_buffer()) ( #62662 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62662
Replaced the methods set_tensor(.) and get_tensor() in the python exposed API from the C++ logic with buffer() and set_buffer(.) to be a cleaner interface.
Reviewed By: SciPioneer
Differential Revision: D30012869
fbshipit-source-id: bd8efab583dd89c96f9aeb3dd48a12073f0b1482
2021-08-04 09:27:31 -07:00
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
Garrett Cramer
4ed2d5d9bb
ps sparse rpc ( #58003 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58003
adds trainer class DdpTrainer
adds trainer class DdpSparseRpcTrainer
adds server class ParameterServerBase
adds server class AverageParameterServer
adds experiment ddp_cpu_sparse_rpc_nccl_allreduce
adds experiment ddp_cuda_sparse_rpc_nccl_allreduce
quip document https://fb.quip.com/iQUtAeKIxWpF
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D29379696
Pulled By: gcramer23
fbshipit-source-id: 9cf5fb7398ba2fa3eb694afbddc4ed00d97f205f
2021-06-24 17:21:49 -07:00