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
..
attention	Allow to run flex_attention on HPU (#148656)	2025-04-16 19:49:15 +00:00
backends
intrinsic
modules	Optimize register_full_backward_hook description when all input no grad (#151785)	2025-04-22 17:57:31 +00:00
parallel	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
qat
quantizable
quantized
utils	[Easy] Optimize clip_grad param description (#151532)	2025-04-17 16:47:38 +00:00
__init__.py
_reduction.py
common_types.py	[5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205)	2025-01-15 04:00:47 +00:00
cpp.py
functional.py	Optimize interpolate saturate description (#151304)	2025-04-17 18:34:29 +00:00
functional.pyi.in	Use Python 3.9 typing (#148157)	2025-03-04 03:09:55 +00:00
grad.py
init.py
parameter.py	remove typo in UninitializedParameter docstring (#140197)	2024-11-15 23:26:23 +00:00
parameter.pyi	remove allow-untyped-defs from torch/nn/parameter.pyi (#144654)	2025-01-13 19:02:31 +00:00