AI/rocm-systems
2
0
Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

update test_profile_general.py

Bladeren bron 
Signed-off-by: Jose Santos <josantos@login1.hpcfund>


[ROCm/rocprofiler-compute commit: ea9ab293e7]
This commit is contained in:
Jose Santos
2024-02-23 16:28:02 -06:00
gecommit door Karl W. Schulz
bovenliggende da18a49f93
commit da3d775fc5
1 gewijzigde bestanden met toevoegingen van 7 en 0 verwijderingen
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files
projects/rocprofiler-compute/tests/test_profile_general.py
+7
Bestand weergeven
@@ -1578,6 +1578,7 @@ def test_mem_levels_HBM():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS
@@ -1608,6 +1609,7 @@ def test_mem_levels_L2():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS
@@ -1638,6 +1640,7 @@ def test_mem_levels_vL1D():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS
@@ -1668,6 +1671,7 @@ def test_mem_levels_LDS():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS
@@ -1698,6 +1702,7 @@ def test_mem_levels_HBM_LDS():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS
@@ -1728,6 +1733,7 @@ def test_mem_levels_vL1D_LDS():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS
@@ -1757,6 +1763,7 @@ def test_mem_levels_L2_vL1D_LDS():
file_dict = test_utils.check_csv_files(workload_dir, num_devices, num_kernels)
if soc == "mi200":
print(sorted(list(file_dict.keys())))
assert sorted(list(file_dict.keys())) == ROOF_ONLY_FILES
else:
assert sorted(list(file_dict.keys())) == ALL_CSVS