Description: Validate and use CUDA graph capture in Megatron Bridge, including local full-iteration graphs and Transformer Engine scoped graphs for attention, MLP, and MoE modules.
Lines: 347 | Code: 88 | Dir: nemo-mbridge-perf-cuda-graphs
name: nemo-mbridge-perf-cuda-graphs description: Validate and use CUDA graph capture in Megatron Bridge, including local full-iteration graphs and Transformer Engine scoped graphs for attention, M…