Reenable tests that now run fine after the IMA fix (#72288)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72288
Fixes https://github.com/pytorch/pytorch/issues/72177 https://github.com/pytorch/pytorch/issues/72189 https://github.com/pytorch/pytorch/issues/72189
These tests were disabled in https://github.com/pytorch/pytorch/pull/72016
Test Plan: Imported from OSS
Reviewed By: ZolotukhinM
Differential Revision: D34009065
Pulled By: anjali411
fbshipit-source-id: 0874d0cc612bb2fb368a092389448b76ac31a989
(cherry picked from commit 5b40a7ba636c74ac1a3cdddc018a272417536ee2)