[PyTorch] [torch.compile] Prepare LayerNormLinear and LayerNormMLP for torch.compile - #42

[PyTorch] [torch.compile] Prepare LayerNormLinear and LayerNormMLP for torch.compile#42

pggPL wants to merge 1 commit into

mainpggPL/TransformerEngine:mainfrom

layernorm_compile_refactorpggPL/TransformerEngine:layernorm_compile_refactorCopy head branch name to clipboard