Add timeout to distributed torch tests to fail on hang#596

ipanfilo merged 4 commits into

devROCm/TransformerEngine:devfrom

ipanfilo/pytorch_tests_timeoutROCm/TransformerEngine:ipanfilo/pytorch_tests_timeoutCopy head branch name to clipboard