fix dimensions in avail output (#82)

* fix dimensions in avail output

* review comment addressed

[ROCm/rocprofiler-sdk commit: 554537f140]
Этот коммит содержится в:
Nagaraj, Sriraksha
2024-12-20 13:26:02 -06:00
коммит произвёл GitHub
родитель 47a66adf1f
Коммит e5dde1b230
2 изменённых файлов: 7 добавлений и 10 удалений
+5 -8
Просмотреть файл
@@ -159,7 +159,7 @@ def get_counters(node_id):
dimensions_stream.write(dim_name)
dimensions_stream.write("[0:")
dimensions_stream.write(str(dim_instance - 1))
dimensions_stream.write(str(dim_instance))
dimensions_stream.write("]")
if dim != no_of_dimensions - 1:
dimensions_stream.write("\t")
@@ -323,12 +323,12 @@ def generate_output(agent_ids):
else:
for node_id in agent_ids:
print(f"gpu-agent:{node_id}\n")
if node_id in agent_basic_counter_map.keys():
basic_counters_stream = io.StringIO()
counters = agent_basic_counter_map[node_id]
for counter in counters:
if counter.block:
basic_counters_stream.write(f"gpu-agent:{node_id}\n")
basic_counters_stream.write("Name:")
basic_counters_stream.write("\t")
basic_counters_stream.write(str(counter.name))
@@ -344,17 +344,16 @@ def generate_output(agent_ids):
basic_counters_stream.write("Dimensions:")
basic_counters_stream.write("\t")
basic_counters_stream.write(str(counter.dimensions))
basic_counters_stream.write("\n")
basic_counters_stream.write("\n\n")
basic_counters = basic_counters_stream.getvalue()
print("List Metrics Basic\n")
print(basic_counters)
print("\n")
basic_counters_stream.close()
if node_id in agent_derived_counter_map.keys():
derived_counters_stream = io.StringIO()
counters = agent_derived_counter_map[node_id]
for counter in counters:
derived_counters_stream.write(f"gpu-agent:{node_id}\n")
derived_counters_stream.write("Name:")
derived_counters_stream.write("\t")
derived_counters_stream.write(str(counter.name))
@@ -370,11 +369,9 @@ def generate_output(agent_ids):
derived_counters_stream.write("Dimensions:")
derived_counters_stream.write("\t")
derived_counters_stream.write(str(counter.dimensions))
derived_counters_stream.write("\n")
derived_counters_stream.write("\n\n")
derived_counters = derived_counters_stream.getvalue()
print("List Metrics Derived\n")
print(derived_counters)
print("\n")
derived_counters_stream.close()
if node_id in agent_pc_sample_config_map.keys():
+2 -2
Просмотреть файл
@@ -198,11 +198,11 @@ iterate_agent_counters_callback(rocprofiler_agent_id_t,
size_t num_counters,
void* user_data)
{
auto* _counters_info = reinterpret_cast<std::vector<std::vector<std::string>>*>(user_data);
auto dimensions_data = std::vector<rocprofiler_record_dimension_info_t>{};
auto* _counters_info = static_cast<std::vector<std::vector<std::string>>*>(user_data);
for(size_t i = 0; i < num_counters; i++)
{
rocprofiler_counter_info_v0_t _info;
auto dimensions_data = std::vector<rocprofiler_record_dimension_info_t>{};
ROCPROFILER_CALL(
rocprofiler_iterate_counter_dimensions(
counters[i], dimensions_info_callback, static_cast<void*>(&dimensions_data)),