From c47c5f224b1d37969d061914b34b860351840a39 Mon Sep 17 00:00:00 2001
From: Akash Kothari <akashk4@tyler.cs.illinois.edu>
Date: Sun, 20 Dec 2020 13:14:20 -0600
Subject: [PATCH] Add script to run all DNNs
---
.../benchmarks/run_dnns_pred.py | 19 +++++++++++++------
1 file changed, 13 insertions(+), 6 deletions(-)
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/run_dnns_pred.py b/llvm/test/VISC/DNN_Benchmarks/benchmarks/run_dnns_pred.py
index 29617bd8e4..4ae6ee5b59 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/run_dnns_pred.py
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/run_dnns_pred.py
@@ -1,13 +1,20 @@
-import os.path
-from os import path
+import os
+import sys
#dnns = "alexnet alexnet2 vgg16_cifar10 vgg16_cifar100 resnet18 mobilenet lenet_mnist"
dnns = "alexnet"
cur_dir = "/home/akashk4/merge/profiling/hpvm/llvm/test/VISC/DNN_Benchmarks/benchmarks"
-if __name__ == "__main__":
- exec_command = "python3 run_dnn_pred.py" + " " + cur_dir + " " + dnns
- print(exec_command)
- os.system(exec_command)
+if sys.argv[1] == "--runtime_tuning":
+ exec_command = "python3 run_dnn_pred.py" + " --runtime_tuning " + cur_dir + " " + dnns
+ print(exec_command)
+ os.system(exec_command)
+elif sys.argv[1] == "--predictive_tuning":
+ exec_command = "python3 run_dnn_pred.py" + " --predictive_tuning " + cur_dir + " " + dnns
+ print(exec_command)
+ os.system(exec_command)
+else:
+ print("Error: Use --runtime_tuning or --predictive_tuning flags!")
+ sys.exit()
--
GitLab