Merge from CTuning (#1198)

mlcommons · Apr 13, 2024 · c3d9b25 · c3d9b25
2 parents 25ab904 + 22ddbf7
commit c3d9b25
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 3 deletions.
diff --git a/cm-mlops/script/app-mlperf-inference-nvidia/_cm.yaml b/cm-mlops/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -262,6 +262,19 @@ deps:
       - run_harness
 
   - tags: get,generic-python-lib,_package.nvmitten,_path./opt/nvmitten-0.1.3-cp38-cp38-linux_x86_64.whl
+    enable_if_env:
+      CM_RUN_STATE_DOCKER:
+        - 'yes'
+        - True
+        - 'True'
+
+  - tags: get,nvidia,mitten
+    skip_if_env:
+      CM_RUN_STATE_DOCKER:
+        - 'yes'
+        - True
+        - 'True'
+
 prehook_deps:
   ########################################################################
   # Install GPTJ-6B model

diff --git a/cm-mlops/script/get-mlperf-inference-src/_cm.json b/cm-mlops/script/get-mlperf-inference-src/_cm.json
@@ -194,7 +194,7 @@
   "versions": {
     "custom": {
       "env": {
-        "CM_MLPERF_LAST_RELEASE": "v3.1"
+        "CM_MLPERF_LAST_RELEASE": "v4.0"
       }
     },
     "deepsparse": {

diff --git a/cm-mlops/script/get-nvidia-mitten/_cm.json b/cm-mlops/script/get-nvidia-mitten/_cm.json
@@ -16,6 +16,9 @@
       ],
       "tags": "get,python3"
     },
+    {
+      "tags": "get,generic-python-lib,_pycuda"
+    },
     {
       "tags": "get,git,_repo.https://github.com/NVIDIA/mitten",
       "force_env_keys": [

diff --git a/cm-mlops/script/get-nvidia-mitten/run.sh b/cm-mlops/script/get-nvidia-mitten/run.sh
@@ -1,3 +1,4 @@
 #!/bin/bash
-
-echo "TBD"
+cd ${CM_NVIDIA_MITTEN_SRC}
+${CM_PYTHON_BIN_WITH_PATH} -m pip install .
+test $? -eq 0 || exit $?