Don't run preprocess for app-mlperf-inference-nvidia inside docker

GATEOverflow · May 20, 2024 · 5fc8591 · 5fc8591
1 parent 7a2af5d
commit 5fc8591
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/script/app-mlperf-inference-nvidia/customize.py b/script/app-mlperf-inference-nvidia/customize.py
@@ -10,6 +10,9 @@ def preprocess(i):
         return {'return':1, 'error': 'Windows is not supported in this script yet'}
     env = i['env']
 
+    if str(env.get('CM_RUN_STATE_DOCKER', '')).lower() in ['1', 'true', 'yes']:
+        return {'return': 0}
+
     if env.get('CM_MODEL', '') == '':
         return {'return': 1, 'error': 'Please select a variation specifying the model to run'}