diff --git a/projects/rocprofiler-compute/src/omniperf b/projects/rocprofiler-compute/src/omniperf index f4b231380d..d483f0bc75 100755 --- a/projects/rocprofiler-compute/src/omniperf +++ b/projects/rocprofiler-compute/src/omniperf @@ -382,7 +382,10 @@ def omniperf_profile(args): # Skip roofline if --no-roof is set. if not args.no_roof: target_binary = detect_roofline() - path_to_binary = str(OMNIPERF_HOME) + "/utils/rooflines/roofline" + "-" + DISTRO_MAP[target_binary["distro"]] + "-" + args.target.lower() + "-rocm" + target_binary["rocm_ver"] + if target_binary["rocm_ver"] == "override": + path_to_binary = target_binary["path"] + else: + path_to_binary = str(OMNIPERF_HOME) + "/utils/rooflines/roofline" + "-" + DISTRO_MAP[target_binary["distro"]] + "-" + args.target.lower() + "-rocm" + target_binary["rocm_ver"] # Distro is valid but cant find rocm ver if not os.path.exists(path_to_binary): print( diff --git a/projects/rocprofiler-compute/src/omniperf_cli/omniperf_cli.py b/projects/rocprofiler-compute/src/omniperf_cli/omniperf_cli.py index 86eebd8ab6..bf900c9d27 100644 --- a/projects/rocprofiler-compute/src/omniperf_cli/omniperf_cli.py +++ b/projects/rocprofiler-compute/src/omniperf_cli/omniperf_cli.py @@ -35,7 +35,6 @@ import sys - import copy import sys import argparse