From a66d3475b63f577f244b196cb9f0740b88a954f1 Mon Sep 17 00:00:00 2001 From: Benjamin Welton Date: Mon, 1 Apr 2024 18:45:04 -0700 Subject: [PATCH] Add additional debug info and do iteration on per-agent basis [ROCm/rocprofiler-sdk commit: 6fc239f6eb71319f710570697ef722e478010817] --- .../tests/counter-collection/validate.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/projects/rocprofiler-sdk/tests/counter-collection/validate.py b/projects/rocprofiler-sdk/tests/counter-collection/validate.py index ceded9efec..93fa53d3b9 100644 --- a/projects/rocprofiler-sdk/tests/counter-collection/validate.py +++ b/projects/rocprofiler-sdk/tests/counter-collection/validate.py @@ -26,13 +26,16 @@ def test_counter_values(input_data): scaling_factor = 1 for itr in data["rocprofiler-sdk-json-tool"]["agents"]: - if itr["type"] == 2 and itr["wave_front_size"] > 0: - scaling_factor = 64 / itr["wave_front_size"] - break + if itr["type"] != 2 or itr["wave_front_size"] < 0: + continue - for itr in data["rocprofiler-sdk-json-tool"]["buffer_records"]["counter_collection"]: - value = itr["counter_value"] - assert int(round(value, 0)) == int(round(1 * scaling_factor, 0)), str(data) + scaling_factor = 64 / itr["wave_front_size"] + + for itr in data["rocprofiler-sdk-json-tool"]["buffer_records"]["counter_collection"]: + value = itr["counter_value"] + assert int(round(value, 0)) == int(round(1 * scaling_factor, 0)), (f"Failure on agent " + f"{str(itr)} expected {1 * scaling_factor} but got {value} " + f"Debug Info:\n {str(data)}") if __name__ == "__main__":