Describe the bug

test_vision_cuda_graphs.py::TestVisionTECudaGraphHelperPP2::test_pp2_create_cudagraphs_first_stage

Steps/Code to reproduce bug

https://github.com/NVIDIA/Megatron-LM/actions/runs/22953339772/job/66628500083

Expected behavior

A clear and concise description of what you expected to happen.

Additional context

Add any other context about the problem here.