From 0be51fc2fefd0ff4246c7ff823c9d3514659d8f6 Mon Sep 17 00:00:00 2001 From: colramos-amd Date: Mon, 22 Jan 2024 16:13:09 -0600 Subject: [PATCH] Catch ROCm-6.0.0 headers for replacement and standardization Signed-off-by: colramos-amd [ROCm/rocprofiler-compute commit: 88ef7b75502ba13d4a15f1b22e063142c81504cd] --- projects/rocprofiler-compute/src/utils/utils.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/projects/rocprofiler-compute/src/utils/utils.py b/projects/rocprofiler-compute/src/utils/utils.py index 9faf92e1d4..403a13acdb 100644 --- a/projects/rocprofiler-compute/src/utils/utils.py +++ b/projects/rocprofiler-compute/src/utils/utils.py @@ -243,8 +243,10 @@ def run_prof(fname, profiler_options, target, workload_dir): # Remove temp directory shutil.rmtree(workload_dir + "/" + "out") - # Overwrite headers + # Standardize rocprof headers via overwrite + # {: } output_headers = { + # ROCm-6.1.0 specific csv headers "KernelName": "Kernel_Name", "Index": "Dispatch_ID", "grd": "Grid_Size", @@ -257,6 +259,12 @@ def run_prof(fname, profiler_options, target, workload_dir): "accum_vgpr": "Accum_VGPR", "BeginNs": "Start_Timestamp", "EndNs": "End_Timestamp", + # ROCm-6.0.0 specific csv headers + "GRD": "Grid_Size", + "WGR": "Workgroup_Size", + "LDS": "LDS_Per_Workgroup", + "SCR": "Scratch_Per_Workitem", + "ACCUM_VGPR": "Accum_VGPR", } df = pd.read_csv(workload_dir + "/" + fbase + ".csv") df.rename(columns=output_headers, inplace=True)