Yuanhao Ji
e3effa5855
Enable UFMT on all of test/distributed ( #123539 )
...
Partially addresses #123062
Ran lintrunner on:
- `test/distributed`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/123539
Approved by: https://github.com/ezyang
2024-04-17 06:46:02 +00:00
PyTorch MergeBot
52be63eb2c
Revert "Enable UFMT on all of test/distributed ( #123539 )"
...
This reverts commit 89ac37fe91https://github.com/pytorch/pytorch/pull/123539 on behalf of https://github.com/DanilBaibak due to Broken trunk ([comment](https://github.com/pytorch/pytorch/pull/123539#issuecomment-2058329471 ))
2024-04-16 06:33:21 +00:00
Yuanhao Ji
89ac37fe91
Enable UFMT on all of test/distributed ( #123539 )
...
Partially addresses #123062
Ran lintrunner on:
- `test/distributed`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/123539
Approved by: https://github.com/ezyang
2024-04-16 03:23:56 +00:00
Sergii Dymchenko
35bf5bac26
Fix "sandcastle_skip_if decorator name is confusing" ( #95649 )
...
Fixes https://github.com/pytorch/pytorch/issues/89473
See the issue https://github.com/pytorch/pytorch/issues/89473
Pull Request resolved: https://github.com/pytorch/pytorch/pull/95649
Approved by: https://github.com/atalman , https://github.com/malfet
2023-03-03 09:29:40 +00:00
Xuehai Pan
046e88a291
[BE] [3/3] Rewrite super() calls in test ( #94592 )
...
Rewrite Python built-in class `super()` calls. Only non-semantic changes should be applied.
- #94587
- #94588
- #94592
Also, methods with only a `super()` call are removed:
```diff
class MyModule(nn.Module):
- def __init__(self):
- super().__init__()
-
def forward(self, ...):
...
```
Some cases that change the semantics should be kept unchanged. E.g.:
f152a79be9/caffe2/python/net_printer.py (L184-L190)f152a79be9/test/test_jit_fuser_te.py (L2628-L2635)https://github.com/pytorch/pytorch/pull/94592
Approved by: https://github.com/ezyang , https://github.com/seemethere
2023-02-12 22:20:53 +00:00
Jeff Daily
72502b94f3
correct use of torch.backends.cudnn.flags() ( #93182 )
...
Fixes #77467 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93182
Approved by: https://github.com/ngimel
2023-01-28 06:50:06 +00:00
Charlie Yan
ffae7308c9
Enable test: distributed/algorithms/quantization/test_quantization ( #80097 )
...
fixes https://github.com/pytorch/pytorch/issues/69017
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80097
Approved by: https://github.com/wanchaol
2022-07-01 01:32:33 +00:00
Wanchao Liang
cf3a5160f8
[BE] move init_multigpu_helper to common_distributed ( #67050 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67050
This PR moves init_multi_gpu_helper to common_distributed so that it could be shared by different distributed tests.
ghstack-source-id: 141370119
Test Plan: wait for ci.
Reviewed By: mrshenli
Differential Revision: D31842644
fbshipit-source-id: c7bad25d6cef9bdce7ad1fb6c60c1cad4b765702
2021-10-22 17:16:11 -07:00
Jane Xu
34051d74da
Add test owner to distributed files starting with test_ ( #66797 )
...
Summary:
Action based on https://github.com/pytorch/pytorch/issues/66232
cc pietern mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse SciPioneer H-Huang
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66797
Reviewed By: gchanan
Differential Revision: D31761389
Pulled By: janeyx99
fbshipit-source-id: c27c9ab4acec1eb71d5edd4538cd113b770dfc6c
2021-10-19 10:55:20 -07:00
Marjan Fariborz
6a76ee04de
Adding alltoall_single collective to collective quantization API ( #63154 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63154
The collective quantization API now supports alltoall, alltoall_single, and allscatter. The test is also included.
ghstack-source-id: 136856877
Test Plan: buck test mode/dev-nosan //caffe2/test/distributed/algorithms/quantization:DistQuantizationTests_nccl -- test_all_to_all_single_bfp16
Reviewed By: wanchaol
Differential Revision: D30255251
fbshipit-source-id: 856f4fa12de104689a03a0c8dc9e3ecfd41cad29
2021-08-27 12:46:31 -07:00
Marjan Fariborz
3b284ab024
Adding BFP16 quantization/dequantization support to OSS ( #63059 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63059
Supporting BFP16 quantization method to OSS. Currently only support CPU
ghstack-source-id: 136639528
Test Plan: Imported from OSS
Reviewed By: wanchaol
Differential Revision: D30194538
fbshipit-source-id: ac248567ad8028457c2a91b77ef2ce81709fce53
2021-08-25 23:41:34 -07:00
Marjan Fariborz
c545b099aa
Separating quantization test from distributed_test ( #63058 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63058
Dedicating separate tests for different quantization methods. Currently supporting FP16 method.
ghstack-source-id: 136499767
Test Plan: uck test mode/dev //caffe2/test/distributed/algorithms/quantization:quantization_gloo_fork -- name_of_the_test
Reviewed By: wanchaol
Differential Revision: D30142580
fbshipit-source-id: 3aacec1a231a662067d2b48c001f0c69fefcdd60
2021-08-24 01:44:55 -07:00
