pytorch

15fef3dc - normalize cuda device (#81739)

normalize cuda device (#81739) Fix for https://github.com/pytorch/pytorch/issues/81068 Pull Request resolved: https://github.com/pytorch/pytorch/pull/81739 Approved by: https://github.com/ezyang