AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix counter collection inconsistency with rocprofv3

Browse Source
This commit is contained in:
Fei Zheng
2025-03-10 21:05:40 -06:00
committed by GitHub
parent 51c9c6fad3
commit 7e8d2d2c0e
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rocprof_compute_soc/soc_base.py
+2 -2
View File
@@ -371,9 +371,9 @@ class CounterFile:
return self.blocks[block].add(counter) return self.blocks[block].add(counter)
# TODO: This is a HACK # FIXME: This is a HACK
def using_v3(): def using_v3():
return "ROCPROF" in os.environ.keys() and os.environ["ROCPROF"] == "rocprofv3" return "ROCPROF" in os.environ.keys() and os.environ["ROCPROF"].endswith("rocprofv3")
@demarcate @demarcate
src/utils/file_io.py
+2 -1
View File
@@ -224,7 +224,8 @@ def create_df_pmc(
dfs.append(tmp_df) dfs.append(tmp_df)
coll_levels.append(f[:-4]) coll_levels.append(f[:-4])
final_df = pd.concat(dfs, keys=coll_levels, axis=1, copy=False) # TODO: double check the case if all tmp_df.shape[0] are not on the same page
final_df = pd.concat(dfs, keys=coll_levels, axis=1, join="inner", copy=False)
if verbose >= 2: if verbose >= 2:
console_debug("pmc_raw_data final_single_df %s" % final_df.info) console_debug("pmc_raw_data final_single_df %s" % final_df.info)
return final_df return final_df