Updates internal `assert_allclose` callsites in favor of `assert_close` (#61841)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61841
Redo of #60863.
Test Plan: Imported from OSS
Reviewed By: ngimel
Differential Revision: D30408145
Pulled By: mruberry
fbshipit-source-id: 0b34ebc7f23ba38ecd89640b61d8aca59b7eab58