[PyTorch] Support distributed weights in GroupedLinear's grouped-tensor path - #3517

[PyTorch] Support distributed weights in GroupedLinear's grouped-tensor path#3517

fanshiqing wants to merge 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

fanshiqing:shiqingf/gtp-grouped-tensor-distributed-weightfanshiqing/TransformerEngine:shiqingf/gtp-grouped-tensor-distributed-weightCopy head branch name to clipboard