pytorch

e3c4cea6 - [functorch] Add support on CUDA keys for control flow ops. (#94465)

[functorch] Add support on CUDA keys for control flow ops. (#94465) Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/94465 Approved by: https://github.com/tugsbayasgalan