Merge branch 'tflite_tpu' of github.com:/makaveli10/ck into tflite_tpu

mlcommons · Aug 14, 2023 · 5d6dd77 · 5d6dd77
2 parents 4193993 + 63a6142
commit 5d6dd77
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/cm-mlops/script/app-mlperf-inference-cpp/_cm.yaml b/cm-mlops/script/app-mlperf-inference-cpp/_cm.yaml
@@ -203,6 +203,8 @@ variations:
 
   retinanet:
     group: model
+    default_env:
+      CM_MLPERF_LOADGEN_MAX_BATCHSIZE: 1
     env:
       CM_MODEL: retinanet
 

diff --git a/cm-mlops/script/app-mlperf-inference-reference/_cm.yaml b/cm-mlops/script/app-mlperf-inference-reference/_cm.yaml
@@ -244,6 +244,16 @@ deps:
     enable_if_env:
       CM_MLPERF_BACKEND:
       - ncnn
+
+  # - tags: get,generic-python-lib
+  #   names:
+  #   - ml-engine-tflite
+  #   enable_if_env:
+  #     CM_MLPERF_BACKEND:
+  #     - tflite
+  #     CM_MLPERF_DEVICE:
+  #     - tpu
+
 
 
   ########################################################################

diff --git a/cm-mlops/script/app-mlperf-inference-reference/customize.py b/cm-mlops/script/app-mlperf-inference-reference/customize.py
@@ -179,6 +179,8 @@ def get_run_cmd_reference(env, scenario_extra_options, mode_extra_options, datas
 
         if env['CM_MLPERF_BACKEND'] == "ncnn":
             env['MODEL_FILE'] = os.path.join(os.path.dirname(env.get('CM_ML_MODEL_FILE_WITH_PATH')), "resnet50_v1")
+        elif env['CM_MLPERF_DEVICE'] == "tpu":
+            env['MODEL_FILE'] = "/mnt/workspace/resnet50_quant_full_mlperf_edgetpu.tflite"
         else:
             env['MODEL_FILE'] = env.get('CM_MLPERF_CUSTOM_MODEL_PATH', env.get('CM_ML_MODEL_FILE_WITH_PATH'))
         if not env['MODEL_FILE']: