compile融合部分算子，提升性能

megatron/core/tensor_parallel/cross_entropy.py

@@ -120,6 +120,7 @@ class VocabParallelCrossEntropy:
 
 
 class _VocabParallelCrossEntropy(torch.autograd.Function):
+    @torch.compile(mode='max-autotune-no-cudagraphs')
     @staticmethod
     def forward(ctx, vocab_parallel_logits, target, label_smoothing=0.0):
         """Vocab parallel cross entropy forward function."""