pytorch

bb2fcc76 - unify TEST_CUDA (#106685)

unify TEST_CUDA (#106685) Fixes #ISSUE_NUMBER as title, unify TEST_CUDA Pull Request resolved: https://github.com/pytorch/pytorch/pull/106685 Approved by: https://github.com/zou3519