Add a checker for roofline.csv in --bottleneck-trace option

Signed-off-by: coleramos425 <colramos@amd.com>
ROCm · Jun 19, 2024 · e4af11f · e4af11f
1 parent a13b9e6
commit e4af11f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/utils/workload_characterization.py b/src/utils/workload_characterization.py
@@ -59,6 +59,15 @@ def parse_omniperf(self):
         perf_df = pd.read_csv(
             os.path.join(self.input_dirs["omniperf"][0][0], "pmc_perf.csv")
         )
+        # Verify that roofline.csv exists
+        if not os.path.exists(
+            os.path.abspath(
+                os.path.join(self.input_dirs["omniperf"][0][0], "roofline.csv")
+            )
+        ):
+            console_error(
+                f"The Omniperf file {os.path.join(self.input_dirs['omniperf'][0][0], 'roofline.csv')} is required to use the --bottleneck-trace flag."
+            )
         roofline_df = pd.read_csv(
             os.path.join(self.input_dirs["omniperf"][0][0], "roofline.csv")
         )