[rocprofv3-avail] - Add sample data (#514)

* Add sample data for avail and remove color code for non terminal output

* review comments

* review comments

* add documentation

* test fix
This commit is contained in:
Nagaraj, Sriraksha
2025-07-22 10:39:59 -05:00
committed by GitHub
parent 57bfd0f607
commit 2447a85215
6 changed files with 46 additions and 8 deletions
+6 -5
View File
@@ -217,7 +217,7 @@ def list_basic_agent(args, list_counters):
agent_info_map = avail.get_agent_info_map()
agent_counters = avail.get_counters()
for agent, info in agent_info_map.items():
for agent, info in dict(sorted(agent_info_map.items())).items():
if (
info["type"] == 2
and args.device is not None
@@ -238,7 +238,7 @@ def list_pc_sampling(args):
sampling_agents = avail.get_pc_sample_configs()
agent_info_map = avail.get_agent_info_map()
print("Agents supporting PC Sampling\n")
for agent in sampling_agents.keys():
for agent in dict(sorted(sampling_agents.items())).keys():
info = agent_info_map[agent]
print(
"{:8}:\t{}\n{:8}:\t{}".format(
@@ -251,8 +251,9 @@ def list_pc_sampling(args):
def info_pc_sampling(args):
sampling_agents = avail.get_pc_sample_configs()
agent_info_map = avail.get_agent_info_map()
for agent, configs in sampling_agents.items():
for agent, configs in dict(sorted(sampling_agents.items())).items():
info = agent_info_map[agent]
print(
"{:8}:\t{}\n{:8}:\t{}".format(
"GPU", info["logical_node_type_id"], "Name", info["name"]
@@ -280,7 +281,7 @@ def listing(args):
agent_counters = avail.get_counters()
agent_info_map = avail.get_agent_info_map()
for agent, info in agent_info_map.items():
for agent, info in dict(sorted(agent_info_map.items())).items():
if (
info["type"] == 2
and args.device is not None
@@ -321,7 +322,7 @@ def info_pmc(args):
print(counter)
print("\n")
for agent, info in agent_info_map.items():
for agent, info in dict(sorted(agent_info_map.items())).items():
if (
info["type"] == 2
and args.device is not None
+6 -1
View File
@@ -1198,7 +1198,7 @@ def run(app_args, args, **kwargs):
if args.marker_trace and not args.suppress_marker_preload:
update_env("LD_PRELOAD", ROCPROF_ROCTX_LIBRARY, append=True)
if trace_count == 0:
if trace_count == 0 and len(app_args) != 0:
warning("No tracing options were enabled.")
# if no tracing was enabled but the options below were enabled, raise an error
@@ -1403,6 +1403,11 @@ def run(app_args, args, **kwargs):
)
else:
app_args = [sys.executable, path, "info", "--pmc"]
for itr in ("ROCPROF", "ROCPROFILER", "ROCTX"):
update_env(
f"{itr}_LOG_LEVEL",
"error",
)
exit_code = subprocess.check_call(
[sys.executable, path, "info", "--pc-sampling"],
env=app_env,