pytorch

1d88635b - Fix cuda-mode and add more tests (#81898)

Fix cuda-mode and add more tests (#81898) Pull Request resolved: https://github.com/pytorch/pytorch/pull/81898 Approved by: https://github.com/ngimel