Enabling GPU device runs for operators (#23461)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23461
Enabling GPU device runs for production operator shapes.
Reviewed By: xw285cornell, mingzhe09088
Differential Revision: D16526928
fbshipit-source-id: 46657963f4b0bc43d14205ccf1b63d588657e388