Added get-aocc script (#240)

* Added get-aocc

Author: arjunsuresh

script/get-aocc/customize.py

@@ -0,0 +1,112 @@
+from mlc import utils
+import os
+
+
+def predeps(i):
+    os_info = i['os_info']
+
+    env = i['env']
+    if env.get('MLC_AOCC_TAR_FILE_PATH', '') != '':
+        env['MLC_AOCC_NEEDS_TAR'] = 'yes'
+
+    return {'return': 0}
+
+
+def preprocess(i):
+
+    os_info = i['os_info']
+
+    env = i['env']
+
+    exe_c = 'clang.exe' if os_info['platform'] == 'windows' else 'clang'
+
+    if env.get('MLC_AOCC_DIR_PATH', '') != '' and env.get(
+            'MLC_AOCC_BIN_WITH_PATH', '') == '':
+        for f in os.listdir(env['MLC_AOCC_DIR_PATH']):
+            if os.path.exists(os.path.join(
+                    env['MLC_AOCC_DIR_PATH'], f, "bin", exe_c)):
+                env['MLC_AOCC_BIN_WITH_PATH'] = os.path.join(
+                    env['MLC_AOCC_DIR_PATH'], f, "bin", exe_c)
+
+    if env.get('MLC_HOST_OS_FLAVOR', '') == 'rhel':
+        if "12" in env.get('MLC_VERSION', '') or "12" in env.get(
+                'MLC_VERSION_MIN', ''):
+            if env.get('MLC_TMP_PATH', '') == '':
+                env['MLC_TMP_PATH'] = ''
+            env['MLC_TMP_PATH'] += "/opt/rh/aocc/root/usr/bin"
+            env['MLC_TMP_PATH_IGNORE_NON_EXISTANT'] = 'yes'
+
+    if 'MLC_AOCC_BIN_WITH_PATH' not in env:
+        r = i['automation'].find_artifact({'file_name': exe_c,
+                                           'env': env,
+                                           'os_info': os_info,
+                                           'default_path_env_key': 'PATH',
+                                           'detect_version': True,
+                                           'env_path_key': 'MLC_AOCC_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):
+    r = i['automation'].parse_version({'match_text': r'AMD\s+clang\sversion\s([\d.]+)',
+                                       'group_number': 1,
+                                       'env_key': 'MLC_AOCC_VERSION',
+                                       'which_env': i['env']})
+    if r['return'] > 0:
+        return r
+    version = r['version']
+
+    print(i['recursion_spaces'] + '    Detected version: {}'.format(version))
+
+    return {'return': 0, 'version': version}
+
+
+def postprocess(i):
+
+    env = i['env']
+    r = detect_version(i)
+    if r['return'] > 0:
+        return r
+
+    env['MLC_COMPILER_FAMILY'] = 'AOCC'
+    version = r['version']
+    env['MLC_COMPILER_VERSION'] = env['MLC_AOCC_VERSION']
+    env['MLC_AOCC_CACHE_TAGS'] = 'version-' + version
+    env['MLC_COMPILER_CACHE_TAGS'] = 'version-' + version + ',family-aocc'
+
+    found_file_path = env['MLC_AOCC_BIN_WITH_PATH']
+
+    found_path = os.path.dirname(found_file_path)
+
+    env['MLC_AOCC_INSTALLED_PATH'] = found_path
+
+    file_name_c = os.path.basename(found_file_path)
+    file_name_cpp = file_name_c.replace('clang', 'clang++')
+    env['FILE_NAME_CPP'] = file_name_cpp
+
+    env['MLC_AOCC_BIN'] = file_name_c
+
+    # General compiler for general program compilation
+    env['MLC_C_COMPILER_BIN'] = file_name_c
+    env['MLC_C_COMPILER_FLAG_OUTPUT'] = '-o '
+    env['MLC_C_COMPILER_WITH_PATH'] = found_file_path
+    env['MLC_C_COMPILER_FLAG_VERSION'] = '--version'
+
+    env['MLC_CXX_COMPILER_BIN'] = file_name_cpp
+    env['MLC_CXX_COMPILER_WITH_PATH'] = os.path.join(found_path, file_name_cpp)
+    env['MLC_CXX_COMPILER_FLAG_OUTPUT'] = '-o '
+    env['MLC_CXX_COMPILER_FLAG_VERSION'] = '--version'
+
+    env['MLC_COMPILER_FLAGS_FAST'] = "-O3"
+    env['MLC_LINKER_FLAGS_FAST'] = "-O3"
+    env['MLC_COMPILER_FLAGS_DEBUG'] = "-O0"
+    env['MLC_LINKER_FLAGS_DEBUG'] = "-O0"
+    env['MLC_COMPILER_FLAGS_DEFAULT'] = "-O2"
+    env['MLC_LINKER_FLAGS_DEFAULT'] = "-O2"
+
+    return {'return': 0, 'version': version}

script/get-aocc/meta.yaml

@@ -0,0 +1,44 @@
+alias: get-aocc
+automation_alias: script
+automation_uid: 5b4e0237da074764
+cache: true
+category: Compiler automation
+clean_files: []
+deps:
+- tags: detect,os
+- tags: extract,file
+  update_tags_from_env_with_prefix:
+    _path.:
+      - MLC_AOCC_TAR_FILE_PATH
+  force_cache: true
+  env:
+    MLC_EXTRACT_FINAL_ENV_NAME: MLC_AOCC_DIR_PATH
+    MLC_EXTRACT_TO_FOLDER: aocc_install
+  enable_if_env:
+    MLC_AOCC_NEEDS_TAR:
+      - yes
+
+input_mapping:
+  tar_file_path: MLC_AOCC_TAR_FILE_PATH
+
+name: Detect or install AOCC compiler
+new_env_keys:
+- MLC_AOCC_*
+- MLC_C_COMPILER_*
+- MLC_CXX_COMPILER_*
+- MLC_COMPILER_*
+- MLC_LINKER_*
+- + CFLAGS
+- + CXXFLAGS
+- + FFLAGS
+- + LDFLAGS
+- +MLC_HOST_OS_DEFAULT_INCLUDE_PATH
+- +PATH
+post_depsq:
+- tags: get,compiler-flags
+sort: 500
+tags:
+- compiler
+- get
+- aocc
+uid: 1ceb0656e99a44ec

script/get-aocc/run.bat

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

script/get-aocc/run.sh

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