frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add deterministic path for CUDA cumsum (#136224)

Browse Source 
Change `cumsum` to call its decomposition when `use_deterministic_algorithms(True)` and input is CUDA.

Fixes #89492

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136224
Approved by: https://github.com/ezyang, https://github.com/justinchuby
This commit is contained in:
Kurt Mohler
2024-09-24 21:34:43 +00:00
committed by PyTorch MergeBot
parent 0133fbcfe7
commit b1a02bf708
9 changed files with 108 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
torch/_tensor_docs.py
View File

@ -1497,15 +1497,6 @@ In-place version of :meth:`~Tensor.cumprod`
""",
)
add_docstr_all(
"cumsum",
r"""
cumsum(dim, dtype=None) -> Tensor
See :func:`torch.cumsum`
""",
)
add_docstr_all(
"cumsum_",
r"""