pytorch/tools/pyi
Kurt Mohler b1a02bf708 Add deterministic path for CUDA cumsum (#136224)
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
2024-09-24 21:34:43 +00:00
..
__init__.py
gen_pyi.py Add deterministic path for CUDA cumsum (#136224) 2024-09-24 21:34:43 +00:00