Updated test to include CK/AITER V2/V3 test in single backend case#454

Micky774 wants to merge 4 commits intodevROCm/TransformerEngine:devfrom

zain/ck-pytorch-testROCm/TransformerEngine:zain/ck-pytorch-testCopy head branch name to clipboard