pytorch

703e83e3 - Fix AARCH64 builds (#121700)

Fix AARCH64 builds (#121700) After https://github.com/pytorch/pytorch/pull/119992 was landed Pull Request resolved: https://github.com/pytorch/pytorch/pull/121700 Approved by: https://github.com/janeyx99, https://github.com/huydhn