Added get-profiler-uprof script

Author: amd-arsuresh

script/get-profiler-uprof/customize.py

@@ -0,0 +1,102 @@
+from mlc import utils
+import os
+from utils import *
+
+
+def ask_url_acceptance(url):
+    print(f"Please take a moment to read the EULA at this URL:\n{url}")
+    print("\nDo you accept the terms of this EULA? [yes/no]")
+
+    while True:
+        response = input().lower()
+        if response in ["yes", "y"]:
+            print("You have accepted the EULA.")
+            return True
+        elif response in ["no", "n"]:
+            print("You have not accepted the EULA.")
+            return False
+        else:
+            print("Invalid input. Please enter 'yes' or 'no'.")
+
+
+def predeps(i):
+    os_info = i['os_info']
+
+    env = i['env']
+    if env.get('MLC_UPROF_TAR_FILE_PATH', '') != '':
+        env['MLC_UPROF_NEEDS_TAR'] = 'yes'
+
+    elif is_true(env.get('MLC_UPROF_DOWNLOAD')) and not is_true(env.get('MLC_UPROF_ACCEPT_EULA')):
+        url = "https://www.amd.com/en/developer/uprof/uprof-eula/uprof-5-1-eula.html?filename=AMDuProf_Linux_x64_5.1.701.tar.bz2"
+        accepted = ask_url_acceptance(url)
+        if accepted:
+            env['MLC_UPROF_ACCEPT_EULA'] = 'yes'
+
+    return {'return': 0}
+
+
+def preprocess(i):
+
+    os_info = i['os_info']
+
+    env = i['env']
+
+    exe = 'AMDuProfSys.exe' if os_info['platform'] == 'windows' else 'AMDuProfSys'
+
+    if 'MLC_UPROF_BIN_WITH_PATH' not in env:
+
+
+        if env.get('MLC_UPROF_DIR_PATH', '') != '':
+            uprof_path = env['MLC_UPROF_DIR_PATH']
+            if os.path.exists(os.path.join(uprof_path, 'bin', exe)):
+                env['MLC_TMP_PATH'] = os.path.join(uprof_path, 'bin')
+
+        r = i['automation'].find_artifact({'file_name': exe,
+                                           'env': env,
+                                           'os_info': os_info,
+                                           'default_path_env_key': 'PATH',
+                                           'detect_version': True,
+                                           'env_path_key': 'MLC_UPROF_BIN_WITH_PATH',
+                                           'run_script_input': i['run_script_input'],
+                                           'recursion_spaces': i['recursion_spaces']})
+        if r['return'] > 0:
+            return r
+
+    return {'return': 0}
+
+
+def detect_version(i):
+    logger = i['automation'].logger
+
+    r = i['automation'].parse_version({'match_text': r'AMDuProfSys version\s+([\d.]+)',
+                                       'group_number': 1,
+                                       'env_key': 'MLC_UPROF_VERSION',
+                                       'which_env': i['env']})
+    if r['return'] > 0:
+        return r
+    version = r['version']
+
+    logger.info(
+        f"{i['recursion_spaces']}    Detected version: {version}")
+
+    return {'return': 0, 'version': version}
+
+
+def postprocess(i):
+
+    env = i['env']
+    r = detect_version(i)
+    if r['return'] > 0:
+        return r
+
+    version = r['version']
+
+    found_file_path = env['MLC_UPROF_BIN_WITH_PATH']
+
+    found_path = os.path.dirname(found_file_path)
+
+    env['MLC_UPROF_INSTALLED_PATH'] = os.path.dirname(found_path)
+
+    env['+PATH'] = [found_path]
+
+    return {'return': 0, 'version': version}

script/get-profiler-uprof/meta.yaml

@@ -0,0 +1,81 @@
+alias: get-profiler-uprof
+automation_alias: script
+automation_uid: 5b4e0237da074764
+cache: true
+category: Compiler automation
+clean_files: []
+deps:
+- tags: detect,os
+- enable_if_env:
+    MLC_UPROF_ACCEPT_EULA:
+    - true
+    MLC_UPROF_URL:
+    - true
+  env:
+    MLC_DOWNLOAD_FINAL_ENV_NAME: MLC_UPROF_TAR_FILE_PATH
+  extra_cache_tags: uprof,download
+  force_cache: true
+  tags: download,file,_mlcutil
+  update_tags_from_env_with_prefix:
+    _url.:
+    - MLC_UPROF_URL
+- enable_if_env:
+    MLC_UPROF_NEEDS_TAR:
+    - true
+  env:
+    MLC_EXTRACT_FINAL_ENV_NAME: MLC_UPROF_DIR_PATH
+    MLC_EXTRACT_USE_SUBDIR_PATH: yes
+    MLC_EXTRACT_TO_FOLDER: uprof_install
+  extra_cache_tags: uprof,extract
+  force_cache: true
+  tags: extract,file
+  update_tags_from_env_with_prefix:
+    _path.:
+    - MLC_UPROF_TAR_FILE_PATH
+input_mapping:
+  accept_eula: MLC_UPROF_ACCEPT_EULA
+  uprof_dir: MLC_UPROF_DIR_PATH
+  tar_file_path: MLC_UPROF_TAR_FILE_PATH
+name: Detect or install AMD uprof
+new_env_keys:
+- +PATH
+- MLC_UPROF_INSTALLED_PATH
+- MLC_UPROF_BIN_WITH_PATH
+post_deps_off:
+- tags: get,compiler-flags
+sort: 500
+tags:
+- get-uprof
+- get
+- uprof
+- uprof-profiler
+tests:
+  run_inputs:
+  - accept_eula: true
+uid: 1530bfcddc3f4fa1
+
+variations:
+  download-and-install:
+    # currently not supported as EULA needs to be approved manually
+    default_version: 5.1
+    env:
+      MLC_UPROF_DOWNLOAD: true
+    group: install-type
+  local-install:
+    default: true
+    env:
+      MLC_UPROF_DOWNLOAD: false
+    group: install-type
+  path.#:
+    default_variations:
+      install-type: local-install
+    env:
+      MLC_UPROF_DIR_PATH: '#'
+
+versions:
+  5.1:
+    env:
+      MLC_UPROF_NEEDS_TAR: true
+      MLC_UPROF_URL: https://download.amd.com/developer/eula/uprof/uprof-5-1/AMDuProf_Linux_x64_5.1.701.tar.bz2
+      MLC_DOWNLOAD_CHECKSUM: a297f370e633505e4c432da4f94c680a
+      MLC_VERSION: 5.1.701

script/get-profiler-uprof/run.bat

@@ -0,0 +1,3 @@
+%MLC_UPROF_BIN_WITH_PATH% --version > tmp-ver.out
+IF %ERRORLEVEL% NEQ 0 EXIT %ERRORLEVEL%
+

script/get-profiler-uprof/run.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+uprof_bin=${MLC_UPROF_BIN_WITH_PATH}
+echo "${uprof_bin} --version"
+
+${uprof_bin} --version > tmp-ver.out
+test $? -eq 0 || exit $?