fix apply_rotary_pos_emb_thd missing cp_group (#146)

Author: hjh0119

src/mcore_bridge/patcher.py

@@ -201,13 +201,14 @@ def _apply_rotary_pos_emb_thd(t: torch.Tensor, cu_seqlens: torch.Tensor, freqs:
             cp_size = cp_group.size()
         else:
             cp_size = mpu.get_context_parallel_world_size()
+            cp_group = mpu.get_context_parallel_group()
         cu_seqlens_for_batched = cu_seqlens // cp_size
         use_batched_rope = (freqs.dim() >= 1 and freqs.shape[0] == cu_seqlens_for_batched[-1]).item()
         # The determination of mla_output_remove_interleaving: a quick solution for identifying deepseek_v4
         # (TODO: refactor)
         if not use_batched_rope and not kwargs.get('mla_output_remove_interleaving', False):
             logger.warning_once('Using non-batched RoPE, which may affect performance.')
-            return _origin_apply_rotary_pos_emb_thd(t, cu_seqlens, freqs, *args, **kwargs)
+            return _origin_apply_rotary_pos_emb_thd(t, cu_seqlens, freqs, *args, cp_group=cp_group, **kwargs)
 
         kwargs.pop('max_seqlen', None)  # compat megatron-lm dev branch
         return rope_utils._apply_rotary_pos_emb_bshd(t.unsqueeze(1), freqs, *args, **kwargs).squeeze(1)