Removing genai_perf collateral

triton-inference-server · Apr 2, 2024 · d41dc56 · d41dc56
1 parent 30be511
commit d41dc56
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/model_analyzer/constants.py b/model_analyzer/constants.py
@@ -71,5 +71,6 @@
 # Model analyzer package name
 PACKAGE_NAME = "triton-model-analyzer"
 
-# GENAI-PERF CSV
+# GENAI-PERF
 GENAI_PERF_CSV = "profile_export_genai_perf.csv"
+GENAI_PERF_COLLATERAL = ["llm_inputs.json", "profile_export.json"]
diff --git a/model_analyzer/perf_analyzer/perf_analyzer.py b/model_analyzer/perf_analyzer/perf_analyzer.py
@@ -29,6 +29,7 @@
 
 from model_analyzer.config.input.config_defaults import DEFAULT_MODEL_TYPE
 from model_analyzer.constants import (
+    GENAI_PERF_COLLATERAL,
     GENAI_PERF_CSV,
     INTERVAL_SLEEP_TIME,
     LOGGER_NAME,
@@ -580,8 +581,8 @@ def _parse_llm_outputs(self, metrics):
             )
 
             os.remove(GENAI_PERF_CSV)
-            os.remove("llm_inputs.json")
-            os.remove("profile_export.json")
+            for filename in GENAI_PERF_COLLATERAL:
+                os.remove(filename)
 
     def _extract_perf_records_from_row(
         self, requested_metrics: List[Record], row_metrics: Dict[str, str]