Bruce Chang a032510db3 shrink_group implementation to expose ncclCommShrink API (#164518) ...

Closes #164529

To expose the new [ncclCommShrink](https://docs.nvidia.com/deeplearning/nccl/user-guide/docs/api/comms.html#ncclcommshrink) API to PyTorch.

This is useful when you need to exclude certain GPUs or nodes from a collective operation, for example in fault tolerance scenarios or when dynamically adjusting resource utilization.

For more info:  [Shrinking a communicator](https://docs.nvidia.com/deeplearning/nccl/user-guide/docs/usage/communicators.html#shrinking-a-communicator)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164518
Approved by: https://github.com/Skylion007, https://github.com/syed-ahmed, https://github.com/kwen2501

2025-10-17 17:55:03 +00:00

..

_internal

shrink_group implementation to expose ncclCommShrink API (#164518)

2025-10-17 17:55:03 +00:00

__init__.py

Add pyrefly suppressions 2/n (#164513)

2025-10-03 02:46:13 +00:00

_comparison.py

Pyrefly suppressions 6/n (#164877)

2025-10-08 02:30:57 +00:00

_creation.py

[BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552)

2025-08-07 00:09:56 +00:00

_utils.py

Fix code descriptions in the test package. (#148145)

2025-03-04 19:14:41 +00:00