[rocprofiler-compute] Fix MI100 tests (#832)

* Fix MI100 tests

* Handle missing roofline in db_analysis.py
This commit is contained in:
vedithal-amd authored and GitHub committed 2025-09-03 11:09:12 -04:00
1 parent baaca0f956
commit 181bdf9ca1
2 files changed
+13 -10

No files matched your search

@@ -403,9 +403,9 @@ class db_analysis(OmniAnalyze_Base):
for workload_path in self._runs.keys():
pmc_df = self._pmc_df_per_workload[workload_path].copy()
sys_info = self._runs[workload_path].sys_info.iloc[0].to_dict()
for key, value in self._roofline_ceilings_per_workload[
workload_path
].items():
for key, value in self._roofline_ceilings_per_workload.get(
workload_path, {}
).items():
sys_info[f"{key}_empirical_peak"] = value
# Calculate PER_XCD variables first
@@ -571,10 +571,12 @@ class db_analysis(OmniAnalyze_Base):
zip(roofline_data_df["Metric"], roofline_data_df["Value"])
)
roofline_data_expressions = {
"total_flops": roofline_data_expressions["Performance (GFLOPs)"],
"l1_cache_data": roofline_data_expressions["AI L1"],
"l2_cache_data": roofline_data_expressions["AI L2"],
"hbm_cache_data": roofline_data_expressions["AI HBM"],
"total_flops": roofline_data_expressions.get(
"Performance (GFLOPs)", ""
),
"l1_cache_data": roofline_data_expressions.get("AI L1", ""),
"l2_cache_data": roofline_data_expressions.get("AI L2", ""),
"hbm_cache_data": roofline_data_expressions.get("AI HBM", ""),
}
roofline_df = pd.DataFrame([