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

Update Roofline output file list

Browse Source 
-  add check device count in check_csv_files
-  reformat

Signed-off-by: JoseSantosAMD <Jose.Santos@amd.com>


[ROCm/rocprofiler-compute commit: 869f9ac8a8]
This commit is contained in:
JoseSantosAMD
2024-01-24 11:52:03 -06:00
committed by Karl W. Schulz
parent 875ab7257e
commit 1a581a453e
2 changed files with 162 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler-compute/tests/test_utils.py
+4 -2
View File
@@ -79,7 +79,7 @@ def clean_output_dir(cleanup, output_dir):
return
def check_csv_files(output_dir, num_kernels):
def check_csv_files(output_dir,num_devices, num_kernels) :
"""Check profiling output csv files for expected number of entries (based on kernel invocations)
Args:
@@ -95,7 +95,9 @@ def check_csv_files(output_dir, num_kernels):
for file in files_in_workload:
if file.endswith(".csv"):
file_dict[file] = pd.read_csv(output_dir + "/" + file)
if not "sysinfo" in file:
if "roofline" in file:
assert len(file_dict[file].index) >= num_devices
elif not "sysinfo" in file:
assert len(file_dict[file].index) >= num_kernels
elif file.endswith(".pdf"):
file_dict[file] = "pdf"