pytorch/torch/_decomp
Ding, Yi1 1d7397a2d0 [Inductor] Avoid tensor slice overflow for large step (#147433)
Fixes #147071

Currently, if step is a value very close to INT64_MAX, the calculation of slice output length will overflow. This PR tries to fix this problem and thus fix #147071.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147433
Approved by: https://github.com/leslie-fang-intel, https://github.com/jansel
2025-03-02 16:07:15 +00:00
..
__init__.py [Inductor] Avoid tensor slice overflow for large step (#147433) 2025-03-02 16:07:15 +00:00
decompositions_for_jvp.py [BE][PYFMT] migrate PYFMT for torch._dynamo to ruff format (#144549) 2025-02-28 03:03:53 +00:00
decompositions_for_rng.py PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102) 2025-01-18 20:47:12 +00:00
decompositions.py [Inductor] Avoid tensor slice overflow for large step (#147433) 2025-03-02 16:07:15 +00:00