mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
fix docstring issues in torch.utils (#113335)
Fixes #112634 Fixes all the issues listed except in `torch/utils/_pytree.py` as the file no longer exists. ### Error counts |File | Count Before | Count now| |---- | ---- | ---- | |`torch/utils/collect_env.py` | 39 | 25| |`torch/utils/cpp_extension.py` | 51 | 13| |`torch/utils/flop_counter.py` | 25 | 8| |`torch/utils/_foreach_utils.py.py` | 2 | 0| |`torch/utils/_python_dispatch.py.py` | 26 | 25| |`torch/utils/backend_registration.py` | 15 | 4| |`torch/utils/checkpoint.py` | 29 | 21| Pull Request resolved: https://github.com/pytorch/pytorch/pull/113335 Approved by: https://github.com/ezyang
This commit is contained in:
committed by
PyTorch MergeBot
parent
44367c59b2
commit
5e10dd2c78
@ -6,15 +6,11 @@ from torch.autograd.grad_mode import no_grad
|
||||
from typing_extensions import TypeAlias
|
||||
|
||||
def _get_foreach_kernels_supported_devices() -> List[str]:
|
||||
r"""
|
||||
Return the device type list that supports foreach kernels.
|
||||
"""
|
||||
r"""Return the device type list that supports foreach kernels."""
|
||||
return ["cuda", "xpu", torch._C._get_privateuse1_backend_name()]
|
||||
|
||||
def _get_fused_kernels_supported_devices() -> List[str]:
|
||||
r"""
|
||||
Return the device type list that supports fused kernels in optimizer.
|
||||
"""
|
||||
r"""Return the device type list that supports fused kernels in optimizer."""
|
||||
return ["cuda", "xpu", torch._C._get_privateuse1_backend_name()]
|
||||
|
||||
TensorListList: TypeAlias = List[List[Optional[Tensor]]]
|
||||
|
Reference in New Issue
Block a user