xref: /aosp_15_r20/external/pytorch/torch/distributed/algorithms/_comm_hooks/__init__.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1from . import default_hooks as default
2
3
4LOW_PRECISION_HOOKS = [
5    default.fp16_compress_hook,
6    default.bf16_compress_hook,
7]
8