search

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.
https://pytorch.org/functorch/
BSD 3-Clause "New" or "Revised" License
1.39k stars 102 forks source link

fix possible overflow #1003

Closed cyyever closed 2 years ago

cyyever commented 2 years ago

Some possible overflows found by a static analyzer

zou3519 commented 2 years ago

@cyyever please submit the change to the functorch folder in the PyTorch repo: https://github.com/pytorch/pytorch/tree/master/functorch

We're keeping this repository (pytorch/functorch) read-only for simplicity. Changes to the functorch in pytorch/pytorch repo will be synced over to the pytorch/functorch repo