[PyTorch] Set flags in norm modules for Mcore sequence-parallel support#1528

timmoon10 merged 1 commit intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

timmoon10:norm-sequence-parallel-flagtimmoon10/TransformerEngine:norm-sequence-parallel-flagCopy head branch name to clipboard