pytorch

2639c4e6 - fix bug in rocm device type (#56646)

fix bug in rocm device type (#56646) Summary: related to https://github.com/pytorch/pytorch/issues/56156. https://github.com/pytorch/pytorch/issues/55808 effectively turned dtypeIfROCM off but let some legacy issues unfixed. Given the fact that we still need to deal with discrepancy between the 2 platform. This PR turns dtypeIfROCM default pointing to dtypeIfCUDA and only override when user specifies. Pull Request resolved: https://github.com/pytorch/pytorch/pull/56646 Reviewed By: mruberry Differential Revision: D27968959 Pulled By: walterddr fbshipit-source-id: 6a11987b8ddf4417577b3d0d5054eaab169de42c