diff --git a/script/add-custom-nvidia-system/meta.yaml b/script/add-custom-nvidia-system/meta.yaml index b07ec8d5e..ec7202857 100644 --- a/script/add-custom-nvidia-system/meta.yaml +++ b/script/add-custom-nvidia-system/meta.yaml @@ -105,7 +105,10 @@ variations: add_deps_recursive: nvidia-inference-common-code: tags: _go - + v5.0: + add_deps_recursive: + nvidia-inference-common-code: + tags: _v5.0 diff --git a/script/app-mlperf-inference-nvidia/meta.yaml b/script/app-mlperf-inference-nvidia/meta.yaml index 86d067c89..1fbba99d6 100644 --- a/script/app-mlperf-inference-nvidia/meta.yaml +++ b/script/app-mlperf-inference-nvidia/meta.yaml @@ -363,10 +363,9 @@ variations: pytorch: tags: _for-nvidia-mlperf-inference-v5.0 pycuda: - version_min: "2024.1" - version: "" + version: "2024.1" nvidia-inference-server: - tags: _mlcommons,_r5.0 + tags: _mlcommons,_v5.0 v4.1: group: version env: diff --git a/script/app-mlperf-inference/meta.yaml b/script/app-mlperf-inference/meta.yaml index c81aa2748..6d796f7ce 100644 --- a/script/app-mlperf-inference/meta.yaml +++ b/script/app-mlperf-inference/meta.yaml @@ -1853,7 +1853,7 @@ variations: reproducibility add_deps_recursive: nvidia-inference-common-code: - tags: _mlcommons,_r5.0 + tags: _mlcommons,_v5.0 nvidia-inference-server: version: r5.0 tags: _mlcommons @@ -1863,6 +1863,8 @@ variations: version: r5.0 nvidia-scratch-space: tags: _version.5.0 + pycuda: + version: "2024.1" default_env: MLC_SKIP_SYS_UTILS: 'yes' MLC_REGENERATE_MEASURE_FILES: 'yes' diff --git a/script/get-mlperf-inference-nvidia-common-code/meta.yaml b/script/get-mlperf-inference-nvidia-common-code/meta.yaml index 27dffb57a..5fda7a3dd 100644 --- a/script/get-mlperf-inference-nvidia-common-code/meta.yaml +++ b/script/get-mlperf-inference-nvidia-common-code/meta.yaml @@ -32,7 +32,7 @@ variations: group: repo-owner nvidia-only: group: repo-owner - r5.0: + v5.0: add_deps_recursive: mlperf-inference-results: tags: _code-only,_v5.0 diff --git a/script/get-mlperf-inference-results/meta.yaml b/script/get-mlperf-inference-results/meta.yaml index a9d90b65b..9b42fdf2a 100644 --- a/script/get-mlperf-inference-results/meta.yaml +++ b/script/get-mlperf-inference-results/meta.yaml @@ -7,7 +7,6 @@ default_env: MLC_GIT_CHECKOUT: master MLC_GIT_DEPTH: --depth 1 MLC_GIT_PATCH: 'no' -default_version: v4.0 deps: [] new_env_keys: - MLC_MLPERF_INFERENCE_RESULTS_* @@ -69,6 +68,7 @@ variations: env: MLC_GIT_URL: https://github.com/<<>>/inference_results_v5.0.git MLC_MLPERF_INFERENCE_RESULTS_VERSION_NAME: v5.0 + MLC_VERSION: "v5.0" versions: v2.1: env: diff --git a/script/get-nvidia-mitten/meta.yaml b/script/get-nvidia-mitten/meta.yaml index 09bac19b9..c5452c480 100644 --- a/script/get-nvidia-mitten/meta.yaml +++ b/script/get-nvidia-mitten/meta.yaml @@ -11,7 +11,8 @@ deps: - python tags: get,python3 - tags: get,generic-python-lib,_pycuda - version: "2023.1" + names: + - pycuda - env: MLC_GIT_CHECKOUT_PATH_ENV_NAME: MLC_NVIDIA_MITTEN_SRC extra_cache_tags: nvidia,mitten,src