Support nvmitten install from src

ctuning · Apr 12, 2024 · e0e6faa · e0e6faa
1 parent d23f1c7
commit e0e6faa
Showing 1 changed file with 13 additions and 0 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