Added lib-mimalloc

Author: amd-arsuresh

script/get-lib-mimalloc/customize.py

@@ -0,0 +1,38 @@
+from utils import *
+from mlc import utils
+import os
+import subprocess
+
+
+def preprocess(i):
+
+    env = i['env']
+    state = i['state']
+
+    os_info = i['os_info']
+
+    cmake_command = f"""{env['MLC_CMAKE_BIN_WITH_PATH']} .. """
+
+    if env.get('MLC_MIMALLOC_CONFIG', '') != '':
+        cmake_command += f""" {env['MLC_MIMALLOC_CONFIG'].replace("'", "")} """
+
+    env['MLC_MIMALLOC_CMAKE_COMMAND'] = cmake_command
+
+    return {'return': 0}
+
+
+def postprocess(i):
+
+    env = i['env']
+    state = i['state']
+
+    os_info = i['os_info']
+
+    lib_path = os.path.join(os.getcwd(), "obj", "lib")
+
+    env['+LD_LIBRARY_PATH'] = [lib_path]
+    env['MLC_MIMALLOC_PATH'] = os.path.dirname(lib_path)
+    env['MLC_MIMALLOC_LIB_PATH'] = lib_path
+    env['MLC_DEPENDENT_CACHED_PATH'] = os.path.join(lib_path, "libmimalloc.so")
+
+    return {'return': 0}

script/get-lib-mimalloc/meta.yaml

@@ -0,0 +1,67 @@
+alias: get-lib-mimalloc
+automation_alias: script
+automation_uid: 5b4e0237da074764
+cache: true
+category: Detection or installation of tools and artifacts
+deps:
+- env:
+    MLC_GIT_CHECKOUT_PATH_ENV_NAME: MLC_MIMALLOC_SRC_PATH
+  extra_cache_tags: mimalloc,repo,mimalloc-repo
+  force_env_keys:
+  - MLC_GIT_*
+  names:
+  - mimalloc-repo
+  tags: get,git,repo
+  update_tags_from_env_with_prefix:
+    _branch.:
+    - MLC_GIT_CHECKOUT
+    _repo.:
+    - MLC_GIT_URL
+    _sha.:
+    - MLC_GIT_SHA
+    _submodules.:
+    - MLC_GIT_SUBMODULES
+- tags: get,cmake
+- env:
+    MLC_GENERIC_SYS_UTIL_IGNORE_MISSING_PACKAGE: true
+  tags: get,generic-sys-util,_build-essential
+new_env_keys:
+- MLC_MIMALLOC_PATH
+- MLC_MIMALLOC_LIB_PATH
+- +LD_LIBRARY_PATH
+new_state_keys: []
+post_deps: []
+posthook_deps: []
+prehook_deps: []
+tags:
+- get-lib-mimalloc
+- get
+- lib
+- mimalloc
+tests:
+  run_inputs:
+  - variations_list:
+    - branch.main
+uid: 81d0dc1269fe41cc
+variations:
+  branch.#:
+    env:
+      MLC_GIT_CHECKOUT: '#'
+  config.#:
+    env:
+      MLC_MIMALLOC_CONFIG: '#'
+  sha.#:
+    env:
+      MLC_GIT_SHA: '#'
+  url.#:
+    env:
+      MLC_GIT_URL: '#'
+    group: version
+  url.official:
+    default: true
+    env:
+      MLC_GIT_URL: https://github.com/microsoft/mimalloc.git
+    group: version
+  version.official:
+    base:
+    - url.official

script/get-lib-mimalloc/run.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+set -e
+
+#Add your run commands here...
+# run "$MLC_RUN_CMD"
+cd ${MLC_MIMALLOC_SRC_PATH}
+mkdir -p obj
+cd obj
+echo "${MLC_MIMALLOC_CMAKE_COMMAND}"
+${MLC_MIMALLOC_CMAKE_COMMAND}
+make