diff --git a/bin/rocprofv2 b/bin/rocprofv2 index d0445e7..6079af8 100755 --- a/bin/rocprofv2 +++ b/bin/rocprofv2 @@ -7,7 +7,8 @@ if [ -n "${ROCP_PRELOAD}" ]; then LD_PRELOAD="${ROCP_PRELOAD}"; fi CURRENT_DIR="$( dirname -- "$0"; )"; ROCPROFV2_DIR=$(dirname -- $(realpath ${BASH_SOURCE[0]})); -ROCM_DIR=$( dirname -- "$ROCPROFV2_DIR"; ) +ROCPROFILER_DIR=$( dirname -- "$ROCPROFV2_DIR"; ) +ROCM_DIR=@rocmtoolkit_merged@ PLUGIN_LIST=("ctf" "perfetto" "file" "att" "cli") RUN_FROM_BUILD=0 if [[ $ROCPROFV2_DIR == *"/build"* ]]; then @@ -15,7 +16,7 @@ if [[ $ROCPROFV2_DIR == *"/build"* ]]; then ROCM_DIR=$ROCPROFV2_DIR fi -export ROCPROFILER_METRICS_PATH=$ROCM_DIR/libexec/rocprofiler/counters/derived_counters.xml +export ROCPROFILER_METRICS_PATH=$ROCPROFILER_DIR/libexec/rocprofiler/counters/derived_counters.xml export LD_LIBRARY_PATH=$ROCM_DIR/lib:$LD_LIBRARY_PATH # Define color code @@ -83,7 +84,7 @@ while [ 1 ]; do exit 1 fi elif [[ "$1" == "--list-counters" ]]; then - export LD_PRELOAD=$LD_PRELOAD:$ROCM_DIR/lib/rocprofiler/librocprofiler_tool.so + export LD_PRELOAD=$LD_PRELOAD:$ROC_DIR/lib/rocprofiler/librocprofiler_tool.so eval $ROCM_DIR/libexec/rocprofiler/ctrl exit 1 elif [[ "$1" == "-i" || "$1" == "--input" ]]; then @@ -221,7 +222,7 @@ while [ 1 ]; do if [ $RUN_FROM_BUILD == 1 ]; then ATT_PATH=$ROCM_DIR/plugin/att/att/att.py else - ATT_PATH=$ROCM_DIR/libexec/rocprofiler/att/att.py + ATT_PATH=$ROCPROFILER_DIR/libexec/rocprofiler/att/att.py export ROCPROFV2_ATT_LIB_PATH=$ROCM_DIR/lib/hsa-amd-aqlprofile/librocprofv2_att.so fi ATT_ARGV=$3 @@ -294,13 +295,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 $* + LD_PRELOAD=$LD_PRELOAD:$ROCPROFILER_DIR/lib/rocprofiler/librocprofiler_tool.so $* 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 $* + LD_PRELOAD=$LD_PRELOAD:$ROCPROFILER_DIR/lib/rocprofiler/librocprofiler_tool.so $* fi get_pmc_results_txt_path() { diff --git a/bin/rpl_run.sh b/bin/rpl_run.sh index 6b236ed..a9c233c 100755 --- a/bin/rpl_run.sh +++ b/bin/rpl_run.sh @@ -25,16 +25,17 @@ ROCPROF_ARGS="$*" time_stamp=`date +%y%m%d_%H%M%S` BIN_DIR=$(dirname $(realpath ${BASH_SOURCE[0]})) -ROOT_DIR=$(dirname $BIN_DIR) +ROCPROFILER_DIR=$(dirname $BIN_DIR) +ROOT_DIR=@rocmtoolkit_merged@ RUN_DIR=`pwd` TMP_DIR="/tmp" DATA_DIR="rpl_data_${time_stamp}_$$" -RPL_PATH=$ROOT_DIR/lib +RPL_PATH=$ROCPROFILER_DIR/lib TLIB_PATH=$RPL_PATH/rocprofiler TTLIB_PATH=$ROOT_DIR/lib/roctracer ROCM_LIB_PATH=$ROOT_DIR/lib -PROF_BIN_DIR=$ROOT_DIR/libexec/rocprofiler +PROF_BIN_DIR=$ROCPROFILER_DIR/libexec/rocprofiler # check if rocprof is supportd on this gpu arch V1_SUPPORTED_GPU_ARCHS=("gfx80x","gfx90x","gfx10xx","gfx94x") @@ -80,7 +81,7 @@ unset ROCP_PROXY_QUEUE # Disable AQL-profile read API export AQLPROFILE_READ_API=0 # ROC Profiler package path -export ROCP_PACKAGE_DIR=$ROOT_DIR +export ROCP_PACKAGE_DIR=$ROCPROFILER_DIR # enabled SPM KFD mode export ROCP_SPM_KFD_MODE=1 @@ -400,7 +401,7 @@ unset_v1_envs() { ################################################################################################ # main -echo "RPL: on '$time_stamp' from '$ROOT_DIR' in '$RUN_DIR'" +echo "RPL: on '$time_stamp' from '$ROCPROFILER_DIR' in '$RUN_DIR'" # Parsing arguments if [ -z "$1" ] ; then usage @@ -633,7 +634,7 @@ elif [ "$input_type" = "txt" -o "$input_type" = "none" ] ; then else echo "" > $RES_DIR/input.xml fi - input_list=`/bin/ls $RES_DIR/input*.xml` + input_list=`ls $RES_DIR/input*.xml` export ROCPROFILER_SESS=$RES_DIR else fatal "Bad input file type '$INPUT_FILE'"