commit 174cb384919d9baa53f668b67e7a160d5907aa5f
author Lei Zhang <antiagainst@google.com> Fri Jul 30 09:54:52 2021 -0400
committer GitHub <noreply@github.com> Fri Jul 30 06:54:52 2021 -0700
tree 800092d47639d5b7c5dccb0ffb38d22aab1a479e
parent 71e709fb7c6d48eee59c79c159756a93706775a7

Fix benchmark model source information (#6606)

Avoid hardcoding "TensorFlow" there; get the source
from the directory name.

build_tools/android/run_benchmarks.py

diff --git a/build_tools/android/run_benchmarks.py b/build_tools/android/run_benchmarks.py
index 557161c..306312b 100755
--- a/build_tools/android/run_benchmarks.py
+++ b/build_tools/android/run_benchmarks.py
@@ -153,9 +153,11 @@
   root_immediate_dir = os.path.basename(benchmark_case_dir)
   iree_driver, target_arch, bench_mode = root_immediate_dir.split("__")
 
+  model_source = os.path.basename(benchmark_category_dir)
+
   return BenchmarkInfo(model_name=model_name,
                        model_tags=model_tags,
-                       model_source="TensorFlow",
+                       model_source=model_source,
                        bench_mode=bench_mode.split(","),
                        runner=iree_driver,
                        device_info=device_info)