Fix another case of float2::x and float2::y may not be the same on ROCm (#35785)
Summary:
This is another case of the issue fixed in https://github.com/pytorch/pytorch/pull/35783. Mirroring 35786.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35785
Differential Revision: D20800317
Pulled By: ezyang
fbshipit-source-id: de5f32839755d5ff5aefff8408df69adbab4d0a1