[https://nvbugs/5442608][fix] Update CUDA graph config for get_model_yaml_config. (NVIDIA#6693)

Signed-off-by: Yuxian Qiu <[email protected]>
Co-authored-by: ruodil <[email protected]>
Signed-off-by: Wangshanshan <[email protected]>

Author: 2 people

tests/integration/defs/perf/pytorch_model_config.py

@@ -159,14 +159,14 @@ def get_model_yaml_config(model_label: str,
                 'llama_v4_maverick_17b_128e_instruct_fp8'
             ],
             'config': {
-                'use_cuda_graph':
-                True,
-                'cuda_graph_padding_enabled':
-                True,
-                'cuda_graph_batch_sizes': [
-                    1, 2, 4, 8, 16, 32, 64, 128, 256, 384, 512, 1024, 2048,
-                    4096, 8192
-                ]
+                'cuda_graph_config': {
+                    'enable_padding':
+                    True,
+                    'batch_sizes': [
+                        1, 2, 4, 8, 16, 32, 64, 128, 256, 384, 512, 1024, 2048,
+                        4096, 8192
+                    ]
+                }
             }
         }
     ]