Fixing bug in capture gpu metrics logic (#819)

nv-braf · web-flow · commit 48449731e741 · 2024-01-24T08:01:00.000-08:00
diff --git a/model_analyzer/record/metrics_manager.py b/model_analyzer/record/metrics_manager.py
@@ -246,7 +246,7 @@ def profile_models(self, run_config: RunConfig) -> Optional[RunConfigMeasurement
             else FileWriter(self._config.perf_output_path)
         )
         capture_gpu_metrics = (
-            run_config.cpu_only() and not self._config.always_report_gpu_metrics
+            self._config.always_report_gpu_metrics or not run_config.cpu_only()
         )
 
         self._print_run_config_info(run_config)

Original file line number	Diff line number	Diff line change
`@@ -246,7 +246,7 @@ def profile_models(self, run_config: RunConfig) -> Optional[RunConfigMeasurement`
`246`	`246`	`else FileWriter(self._config.perf_output_path)`
`247`	`247`	`)`
`248`	`248`	`capture_gpu_metrics = (`
`249`		`- run_config.cpu_only() and not self._config.always_report_gpu_metrics`
	`249`	`+ self._config.always_report_gpu_metrics or not run_config.cpu_only()`
`250`	`250`	`)`
`251`	`251`
`252`	`252`	`self._print_run_config_info(run_config)`