Marjan Fariborz c7db642a72 Adding collective quantization API (#62142) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62142 Created wrapper that takes the collective op and a quantization type as an arguments. It quantize the input, performs the collective op, and and perform dequantization Test Plan: Tested through distributed_gloo_fork. e.g., buck test mode/dev-nosan caffe2/test/distributed:distributed_nccl_fork -- test_all_to_all_quantized Reviewed By: wanchaol Differential Revision: D29682812 fbshipit-source-id: 79c39105ff11270008caa9f566361452fe82a92e		2021-08-09 08:11:22 -07:00
..
ddp_comm_hooks	[DDP Communication Hook] Update get_tensor and set_tensor to be cleaner naming conventions (buffer() and set_buffer()) (#62662)	2021-08-04 09:27:31 -07:00
model_averaging	[Model Averaging] Fix docstring of PeriodicModelAverager (#62392)	2021-07-29 17:26:27 -07:00
__init__.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00
join.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00
quantization.py	Adding collective quantization API (#62142)	2021-08-09 08:11:22 -07:00