From 91ccbe97ffbb025da570f8af16b0cd3a1f5c7c4a Mon Sep 17 00:00:00 2001
From: gobhardw <gopesh.bhardwaj@amd.com>
Date: Wed, 27 Mar 2024 13:00:44 +0530
Subject: [PATCH] Fixing llama issue for arg parsing

Change-Id: I9e3e4a37493a7f280b1ef8a4c6ec6d940ddae6ce
---
 bin/rocprofv2 | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/bin/rocprofv2 b/bin/rocprofv2
index b8779bed44..f949eb84c2 100755
--- a/bin/rocprofv2
+++ b/bin/rocprofv2
@@ -17,6 +17,7 @@ fi
 
 export ROCPROFILER_METRICS_PATH=$ROCM_DIR/libexec/rocprofiler/counters/derived_counters.xml
 export LD_LIBRARY_PATH=$ROCM_DIR/lib:$LD_LIBRARY_PATH
+export HSA_TOOLS_LIB=$ROCM_DIR/lib/librocprofiler64.so.2
 
 # Define color code
 GREEN='\033[0;32m'
@@ -282,6 +283,8 @@ if [ -z $ROCPROFILER_PLUGIN_LIB ]; then
   fi
 fi
 
+export LD_PRELOAD=$LD_PRELOAD:$ROCM_DIR/lib/rocprofiler/librocprofiler_tool.so
+
 if [ -n "$PMC_LINES" ] && [ ! -n "$ATT_ARGV" ]; then
   COUNTER=1
   for i in ${!PMC_LINES[@]}; do
@@ -295,13 +298,13 @@ if [ -n "$PMC_LINES" ] && [ ! -n "$ATT_ARGV" ]; then
       export OUTPUT_PATH=$FINAL_PATH
     fi
     let COUNTER=COUNTER+1
-    LD_PRELOAD=$LD_PRELOAD:$ROCM_DIR/lib/rocprofiler/librocprofiler_tool.so $*
+    exec "${@}"
     if [ -n "$OUTPUT_PATH" ]; then
       echo -e "\nThe output path for the following counters: $OUTPUT_PATH"
     fi
   done
 else
-  LD_PRELOAD=$LD_PRELOAD:$ROCM_DIR/lib/rocprofiler/librocprofiler_tool.so $*
+  exec "${@}"
 fi
 
 get_pmc_results_txt_path() {