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
replace.vim
tensor.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
utils.py