Add LookupModel() to find a model in the model cache without creating it.

Signed-off-by: Adrian Hunter <[email protected]>
---
 tools/perf/scripts/python/exported-sql-viewer.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tools/perf/scripts/python/exported-sql-viewer.py 
b/tools/perf/scripts/python/exported-sql-viewer.py
index 61b3911d91e6..18ad04654adc 100755
--- a/tools/perf/scripts/python/exported-sql-viewer.py
+++ b/tools/perf/scripts/python/exported-sql-viewer.py
@@ -341,6 +341,15 @@ def LookupCreateModel(model_name, create_fn):
        model_cache_lock.release()
        return model
 
+def LookupModel(model_name):
+       model_cache_lock.acquire()
+       try:
+               model = model_cache[model_name]
+       except:
+               model = None
+       model_cache_lock.release()
+       return model
+
 # Find bar
 
 class FindBar():
-- 
2.17.1

Reply via email to