[PyTorch] Refactor grouped linear and grouped MLP tests#3122

timmoon10 merged 6 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

timmoon10:tmoon/refactor-grouped-mlp-tests2timmoon10/TransformerEngine:tmoon/refactor-grouped-mlp-tests2Copy head branch name to clipboard