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 29617bd8e43dea35b8333dbcbcd952ea5f5df183..4ae6ee5b595c6d2becf9b0dea0c2db3fc1e0c805 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()