[ROCm] Don't use MIOpen for tensors with more than INT_MAX number of elements (#37110)
Summary:
This pull request extends the fallback implemented in https://github.com/pytorch/pytorch/issues/31383 to not use MIOpen for tensors where number of elements in a tensor exceeds INT_MAX. The PR also enables the corresponding test in TestNN
cc: ezyang jeffdaily
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37110
Differential Revision: D21196336
Pulled By: ezyang
fbshipit-source-id: 25fd80308a0e2f7941c249735674ebc85d3fd39e