Yu, Guangye 3d3fcaaf7b Delegate torch.accelerator.device_count to torch.xxx.device_count for multi-process usage (#149924) ... # Motivation Adapt `torch.accelerator.device_count` for multi-process usage. For example, `torch.cuda.device_count` avoids poisoning fork, then `torch.accelerator.device_count` should meet the same requirement. Now that `torch.get_device_module(device).device_count` supports this, `torch.accelerator.device_count` should align with this behavior as well. Pull Request resolved: https://github.com/pytorch/pytorch/pull/149924 Approved by: https://github.com/albanD ghstack dependencies: #147507		2025-04-10 02:37:37 +00:00
..
__init__.py	Delegate torch.accelerator.device_count to torch.xxx.device_count for multi-process usage (#149924)	2025-04-10 02:37:37 +00:00
_utils.py	[Accelerator][Chore] Use existing acc when raising an error (#150829)	2025-04-08 16:05:06 +00:00