[megatron] Fix TE sliding-window attention crash at micro_batch>1 on the non-packed forward path#1848

erictang000 merged 2 commits into

NovaSky-AI:mainNovaSky-AI/SkyRL:mainfrom

john-hahn:john/megatron-te-swa-maskjohn-hahn/SkyRL:john/megatron-te-swa-maskCopy head branch name to clipboard