48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import pytest
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def test_json_data(json_data):
|
||
|
|
data = json_data["rocprofiler-sdk-tool"]
|
||
|
|
strings = data["strings"]
|
||
|
|
assert "att_filenames" in strings.keys()
|
||
|
|
att_files = data["strings"]["att_filenames"]
|
||
|
|
assert len(att_files) > 0
|
||
|
|
|
||
|
|
|
||
|
|
def test_code_object_memory(code_object_file_path, json_data, output_path):
|
||
|
|
|
||
|
|
data = json_data["rocprofiler-sdk-tool"]
|
||
|
|
tool_memory_load = data["strings"]["code_object_snapshot_filenames"]
|
||
|
|
gfx_pattern = "gfx[a-z0-9]+"
|
||
|
|
match = re.search(gfx_pattern, tool_memory_load[0])
|
||
|
|
assert match != None
|
||
|
|
gpu_name = match.group(0)
|
||
|
|
tool_memory_load_1 = open(os.path.join(output_path, tool_memory_load[0]), "rb")
|
||
|
|
tool_memory_load_2 = open(os.path.join(output_path, tool_memory_load[1]), "rb")
|
||
|
|
found = False
|
||
|
|
for hsa_file in code_object_file_path["hsa_memory_load"]:
|
||
|
|
|
||
|
|
m = re.search(gfx_pattern, hsa_file)
|
||
|
|
assert m != None
|
||
|
|
gpu = m.group(0)
|
||
|
|
|
||
|
|
if gpu == gpu_name:
|
||
|
|
found = True
|
||
|
|
hsa_memory_load = open(hsa_file, "rb")
|
||
|
|
hsa_memory_fs = hsa_memory_load.read()
|
||
|
|
tool_memory_fs_1 = tool_memory_load_1.read()
|
||
|
|
tool_memory_fs_2 = tool_memory_load_2.read()
|
||
|
|
assert hsa_memory_fs == tool_memory_fs_2 or hsa_memory_fs == tool_memory_fs_1
|
||
|
|
break
|
||
|
|
assert found == True
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
exit_code = pytest.main(["-x", __file__] + sys.argv[1:])
|
||
|
|
sys.exit(exit_code)
|