add more generic support for distributed trainer tests (#46109)
* add more generic support for distributed trainer tests
Signed-off-by: kaixuan1992 <kaixuan.liu92@gmail.com>
* update
Signed-off-by: kaixuan1992 <kaixuan.liu92@gmail.com>
---------
Signed-off-by: kaixuan1992 <kaixuan.liu92@gmail.com>
Co-authored-by: kaixuan1992 <kaixuan.liu92@gmail.com>