diff --git a/data_axs.json b/data_axs.json index 6a875fe..ba9a5f5 100644 --- a/data_axs.json +++ b/data_axs.json @@ -36,8 +36,7 @@ "base_power_measurement_experiment": "base_power_measurement_experiment", "text_to_image_reference_using_torch_loadgen": "text_to_image_reference_using_torch_loadgen", "cnndm_preprocessor": "cnndm_preprocessor", - "docker_qaic": "docker_qaic", - "docker_trtllm": "docker_trtllm" + "docker_qaic": "docker_qaic" }, "repo_name": "axs2mlperf", "submodules": false diff --git a/docker_trtllm/code_axs.py b/docker_trtllm/code_axs.py deleted file mode 100644 index 7cc9810..0000000 --- a/docker_trtllm/code_axs.py +++ /dev/null @@ -1,26 +0,0 @@ -import getpass -import socket - -def get_user_name(): - return getpass.getuser() - -def get_hostname(): - return socket.gethostname() - -def create_mounted_string(local_dir, container_local_dir, repo_list, local_experiments_dir, local_experiment_dir_name, source_dir, code_dir): - result = "" - for elem in repo_list: - if elem != local_experiment_dir_name: - end = "_main" - else: - end = "" - if elem == local_experiment_dir_name: - local_dir = local_experiments_dir - - result += " -v " + local_dir + elem + ":" + container_local_dir + elem + end - result += " -v " + source_dir + ":" + code_dir - return result - -def create_ccache_dirs(code_dir, cache_dir): - result = " --env \"CCACHE_DIR=" + code_dir + cache_dir + "\"" + " --env \"CCACHE_BASEDIR=" + code_dir + "\"" + " " - return result diff --git a/docker_trtllm/data_axs.json b/docker_trtllm/data_axs.json deleted file mode 100644 index bca2d6e..0000000 --- a/docker_trtllm/data_axs.json +++ /dev/null @@ -1,47 +0,0 @@ -{ - "_parent_entries": [ - [ "^", "byname", "shell" ] - ], - "_producer_rules": [ - [ [ "container_trtllm_loadgen_output" ], ["run"], { - } ] - ], - "user": [ "^^", "get_user_name" ], - "hostname": [ "^^", "get_hostname" ], - - "source_dir": "/home/oleg/TensorRT-LLM", - "code_dir": "/code/tensorrt_llm", - "cache_dir": "/cpp/.ccache", - - "container_name": [ "^^", "substitute", "axs_#{user}#" ], - "container_local_dir": "/home/krai/work_collection/", - - "local_dir": [ "^^", "substitute", "/home/#{user}#/work_collection/" ], - "local_experiment_dir_name": "experiments", - "local_experiments_dir": [ "^^", "substitute", "/home/#{user}#/work_collection/" ], - - "container_id": [ "^^", "substitute", "`if [ $(docker ps -qf \"name=#{container_name}#\" | head -c1 | wc -c) -ne 0 ]; then docker ps -qf \"name=#{container_name}#\"; else docker run -dt --name #{container_name}# --ipc=host --ulimit memlock=#{memlock}# --ulimit stack=#{stack}# --gpus=#{gpus}# #{mounted_repos_str}# #{ccache_dirs}# --workdir #{code_dir}# --hostname #{hostname}# --tmpfs #{tmpfs}# --rm #{image_name}#:#{image_tag}# bash; fi;`" ], - - "repo_list": [ "axs2tensorrt-dev", "experiments" ], - "mounted_repos_str": [ "^^", "create_mounted_string" ], - "ccache_dirs": [ "^^", "create_ccache_dirs" ], - - "ccache_dir": "/code/tensorrt_llm/cpp/.ccache", - "ccache_basedir": "/code/tensorrt_llm", - - "image_name": "tensorrt_llm/release", - "image_tag": "latest", - - "container_only": false, - - "cmd": [ "^^", "case", [ [ "^^", "get", "container_only" ], - true, [ "^^","substitute", "export CONTAINER_ID=#{container_id}# ; echo \"CONTAINER_ID = `echo ${CONTAINER_ID} | cut -c1-12`\"" ]] , - { "default_value": [ "^^","substitute", " docker exec #{container_id}# bash -c \"axs byquery #{modified_query}#\""] } - ], - - "modified_query": [ "^^", "execute", [[ - [ "get", "__query" ], - [ "replace", ["container_trtllm_loadgen_output", "loadgen_output"] ] - ]] ], - "shell_cmd_with_subs": "#{cmd}#" -}