From f5505b598925960bca1804bfd1aa29c055dd3cae Mon Sep 17 00:00:00 2001 From: vedithal-amd Date: Wed, 24 Sep 2025 10:31:20 -0400 Subject: [PATCH] Use ROCM_PATH for sdk library path (#1097) --- projects/rocprofiler-compute/src/argparser.py | 5 ++++- projects/rocprofiler-compute/tests/conftest.py | 6 +++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/projects/rocprofiler-compute/src/argparser.py b/projects/rocprofiler-compute/src/argparser.py index 2445d716f7..e625b04d02 100644 --- a/projects/rocprofiler-compute/src/argparser.py +++ b/projects/rocprofiler-compute/src/argparser.py @@ -24,6 +24,7 @@ ############################################################################## import argparse +import os import re from pathlib import Path from typing import Optional @@ -351,7 +352,9 @@ Examples: type=str, dest="rocprofiler_sdk_library_path", required=False, - default="/opt/rocm/lib/librocprofiler-sdk.so", + default=str( + Path(os.getenv("ROCM_PATH", "/opt/rocm")) / "lib/librocprofiler-sdk.so" + ), help="\t\t\tSet the path to rocprofiler SDK library.", ) profile_group.add_argument( diff --git a/projects/rocprofiler-compute/tests/conftest.py b/projects/rocprofiler-compute/tests/conftest.py index db2cb368bc..66972235b0 100644 --- a/projects/rocprofiler-compute/tests/conftest.py +++ b/projects/rocprofiler-compute/tests/conftest.py @@ -23,8 +23,10 @@ ############################################################################## +import os import subprocess from importlib.machinery import SourceFileLoader +from pathlib import Path from unittest.mock import patch import pytest @@ -45,7 +47,9 @@ def pytest_addoption(parser): parser.addoption( "--rocprofiler-sdk-library-path", type=str, - default="/opt/rocm/lib/librocprofiler-sdk.so", + default=str( + Path(os.getenv("ROCM_PATH", "/opt/rocm")) / "lib/librocprofiler-sdk.so" + ), help="Path to the rocprofiler-sdk library", )