AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[rocprofv3] Fix ATT library path (#476)

Browse Source 
* Fix library path

* Update docs

* Review comments

* Update source/bin/rocprofv3.py

---------

Co-authored-by: Giovanni Baraldi <gbaraldi@amd.com>
This commit is contained in:
Baraldi, Giovanni
2025-07-01 22:08:29 +02:00
committed by GitHub
parent 1524ceb84d
commit c0c08b2f08
2 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/bin/rocprofv3.py
+12 -6
View File
@@ -166,7 +166,7 @@ def resolve_library_path(val, args, is_sdk_lib=True):
return val
def check_att_capability(args, att_lib_name="librocprof-trace-decoder.so"):
def get_att_paths(args):
ROCPROFV3_DIR = os.path.dirname(os.path.realpath(__file__))
ROCM_DIR = os.path.dirname(ROCPROFV3_DIR)
@@ -177,17 +177,23 @@ def check_att_capability(args, att_lib_name="librocprof-trace-decoder.so"):
if args.att_library_path:
library_paths.extend(args.att_library_path)
elif os.environ.get("ROCPROF_ATT_LIBRARY_PATH"):
return os.environ.get("ROCPROF_ATT_LIBRARY_PATH")
else:
default_lib_path_env = os.environ.get("LD_LIBRARY_PATH", "").split(":") + [
f"{ROCM_DIR}/lib"
]
defined_att_path_env = os.environ.get("ROCPROF_ATT_LIBRARY_PATH", "").split(":")
att_paths = defined_att_path_env if defined_att_path_env else default_lib_path_env
for itr in att_paths:
# don't add duplicates
for itr in default_lib_path_env:
if itr not in library_paths:
library_paths += [itr]
return library_paths
def check_att_capability(args, att_lib_name="librocprof-trace-decoder.so"):
library_paths = get_att_paths(args)
for path in library_paths:
for root, dirs, files in os.walk(path, topdown=True):
for itr in files:
@@ -1550,7 +1556,7 @@ def run(app_args, args, **kwargs):
)
else:
fatal_error(
f"rocprof-trace-decoder library path not found in {args.att_library_path}"
f"rocprof-trace-decoder library path not found in {get_att_paths(args)}"
)
if args.att_perfcounters: