make device normalization more generic in faketensor (#102519)
Fixes #ISSUE_NUMBER
make the device normalization more generic in faketensor to support devices like "cuda", "foo" and so on.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/102519
Approved by: https://github.com/albanD
