Gopesh Bhardwaj
34 wiersze
917 B
Python
34 wiersze
917 B
Python
import sys
|
|
import pytest
|
|
|
|
|
|
def test_validate_counter_collection_plus_tracing(
|
|
json_data, counter_input_data, hsa_input_data
|
|
):
|
|
|
|
# check if either kernel-name/FUNCTION is present
|
|
assert (
|
|
"Kernel_Name" in counter_input_data.columns
|
|
or "Function" in counter_input_data.columns
|
|
)
|
|
|
|
data = json_data["rocprofiler-sdk-tool"]
|
|
hsa_api = data["buffer_records"]["hsa_api"]
|
|
assert len(hsa_input_data) == len(hsa_api)
|
|
|
|
counter_collection_data = data["callback_records"]["counter_collection"]
|
|
assert len(counter_collection_data) > 0
|
|
|
|
|
|
def test_perfetto_data(pftrace_data, json_data):
|
|
import rocprofiler_sdk.tests.rocprofv3 as rocprofv3
|
|
|
|
rocprofv3.test_perfetto_data(
|
|
pftrace_data, json_data, ("hip", "hsa", "marker", "kernel", "memory_copy")
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
exit_code = pytest.main(["-x", __file__] + sys.argv[1:])
|
|
sys.exit(exit_code)
|