Change config param cuda_graph to enable_cuda_graph

Signed-off-by: Vasudevan Rengasamy <[email protected]>

nemo/collections/nlp/models/language_modeling/megatron/gpt_full_te_layer_autocast_spec.py

@@ -262,7 +262,7 @@ def forward(
         context = None
 
         # CUDA graph requires returned values to be Tensors
-        if self.config.cuda_graph and self.training:
+        if self.config.enable_cuda_graph and self.training:
             return hidden_states
         return hidden_states, context