Yu, Guangye ad81eeb7c7 Refactor to use torch.accelerator.device_index instead of torch.cuda.device for generic device context manager (#148880) ...

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148880
Approved by: https://github.com/EikanWang, https://github.com/albanD
ghstack dependencies: #148864

2025-04-25 09:45:25 +00:00

..

__init__.py

remove allow-untyped-defs for torch/nn/parallel/__init__.py (#143437)

2024-12-18 08:50:37 +00:00

_functions.py

Refactor to use torch.accelerator.device_index instead of torch.cuda.device for generic device context manager (#148880)

2025-04-25 09:45:25 +00:00

comm.py

[5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205)

2025-01-15 04:00:47 +00:00

data_parallel.py

[5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205)

2025-01-15 04:00:47 +00:00

distributed.py

[DDP] add one option to allow skipping all reduce unused parameters (#151503)

2025-04-17 23:30:19 +00:00

parallel_apply.py

[5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205)

2025-01-15 04:00:47 +00:00

replicate.py

[5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205)

2025-01-15 04:00:47 +00:00

scatter_gather.py

[5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205)

2025-01-15 04:00:47 +00:00