From 3a9fcb1c0e2371b5394046e8122f108717e7b53f Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Sat, 22 Feb 2025 11:24:41 +0000
Subject: [PATCH] Fix for no-cache in run-mobilenets (#256)

* Fix no cache issue for mobilenet

* Added test for mobilenet
---
 .../run-mlperf-inference-mobilenet-models/customize.py |  1 +
 script/run-mlperf-inference-mobilenet-models/meta.yaml | 10 ++++++++++
 2 files changed, 11 insertions(+)

diff --git a/script/run-mlperf-inference-mobilenet-models/customize.py b/script/run-mlperf-inference-mobilenet-models/customize.py
index 6641448e2..4e0641637 100644
--- a/script/run-mlperf-inference-mobilenet-models/customize.py
+++ b/script/run-mlperf-inference-mobilenet-models/customize.py
@@ -17,6 +17,7 @@ def preprocess(i):
     adr = i['input'].get('adr')
 
     automation = i['automation']
+    mlc = i['automation'].action_object
 
     quiet = (env.get('MLC_QUIET', False) == 'yes')
     verbose = (env.get('MLC_VERBOSE', False) == 'yes')
diff --git a/script/run-mlperf-inference-mobilenet-models/meta.yaml b/script/run-mlperf-inference-mobilenet-models/meta.yaml
index 93d8ecc17..b1d2b3f0c 100644
--- a/script/run-mlperf-inference-mobilenet-models/meta.yaml
+++ b/script/run-mlperf-inference-mobilenet-models/meta.yaml
@@ -152,3 +152,13 @@ variations:
     group: optimization
   use-neon:
     alias: neon
+tests:
+  run_inputs:
+  - env:
+      MLC_TEST_ONE_RUN: 'yes'
+    variations_list: 
+    - tflite
+    - accuracy_only
+    adr:
+      compiler:
+        tags: gcc