Clean up PyTorch FA tests#427

wangye805 merged 2 commits intodevROCm/TransformerEngine:devfrom

zain/pytorch-fa-test-cleanupROCm/TransformerEngine:zain/pytorch-fa-test-cleanupCopy head branch name to clipboard