Split ASAN and ROCM tests into test1 and test2 (#41520)
Summary:
This should reduce end-to-end test runtime for 2 slowest configs
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41520
Reviewed By: seemethere
Differential Revision: D22575028
Pulled By: malfet
fbshipit-source-id: a65bfa5932fcda3cf0f4fdd97bcc7ebb3f54c281