Integrate CUDA Graph 4D in test_megatron_e2e_pipeline.py#121

GindaChen wants to merge 5 commits into

mainhao-ai-lab/DistCA:mainfrom

junda/cuda-graph-fix-tphao-ai-lab/DistCA:junda/cuda-graph-fix-tpCopy head branch name to clipboard