diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet2_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet2_promise.cc
index 2cd364bf6b184aa271d9916fc6900871a57bd039..a7333abd99aeed26b47d56eff2af9a5d5e8314bb 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet2_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet2_promise.cc
@@ -39,11 +39,16 @@ int main(int argc, char* argv[]){
   if(argc > 5){
     offset = atoi(argv[5]);   
   }
+  
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+
 
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc
index 81cf894f434f0b34c7a74d98ef66a24fcb633056..3026f5e57391bfbf07df50ce05b1b8be2919a866 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/alexnet_promise.cc
@@ -41,10 +41,14 @@ int main(int argc, char* argv[]){
     offset = atoi(argv[5]);   
   }
 
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/lenet_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/lenet_promise.cc
index ff1fcba79cd96f213181ba275542018fb036b57e..41451598c514bd8ef5c2f62a5f0c5f33b6aab0ab 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/lenet_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/lenet_promise.cc
@@ -141,8 +141,11 @@ int main(int argc, char* argv[]){
     offset = atoi(argv[5]);   
   }
 
-
   if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+  
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_promise.cc
index 691c97a3ef4159fd61df5bcabf8adc14fd28eabd..47c1d01af96ad2d4bd6eca9c13f32a6d7b1fb829 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_promise.cc
@@ -41,10 +41,15 @@ int main(int argc, char* argv[]){
     offset = atoi(argv[5]);   
   }
 
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+  
+  
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_shallow_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_shallow_promise.cc
index 89b3697298f896c0975b7d6e458e88fe7db96ca1..120323e607e3571df395003a770925b09da2a65e 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_shallow_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/mobilenet_shallow_promise.cc
@@ -41,10 +41,15 @@ int main(int argc, char* argv[]){
     offset = atoi(argv[5]);   
   }
 
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+  
+  
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/resnet18_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/resnet18_promise.cc
index 21c7b051c24c9c053041f6cf4d63e8ace3b3645d..261753c1a98a1bb000f842a2cdee406c6a6fa7a3 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/resnet18_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/resnet18_promise.cc
@@ -40,12 +40,16 @@ int main(int argc, char* argv[]){
   if(argc > 5){
     offset = atoi(argv[5]);   
   }
+  
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
 
   
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar100_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar100_promise.cc
index 27bc38e0e336d970f3f3096bdcc447e29e8b9a33..0918bb6855cf97eefc5b7be5cbd22d31fbf5cc9b 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar100_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar100_promise.cc
@@ -43,11 +43,15 @@ int main(int argc, char* argv[]){
     offset = atoi(argv[5]);   
   }
 
-
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+  
+  
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar10_promise.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar10_promise.cc
index c66de402d86f406b141738b2c0652c5084af81b6..2fb5193ce0ee5c3daa2ceca0eba163134b452ac7 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar10_promise.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/promise/vgg16_cifar10_promise.cc
@@ -42,11 +42,15 @@ int main(int argc, char* argv[]){
     offset = atoi(argv[5]);   
   }
 
+  if(argc > 6){
+    batch_size = atoi(argv[6]);   
+  }
+  
  
   bool shouldDumpClassConf = false;
   float* classConfs;
   int* predictedLabels;
-  if(argc > 6){
+  if(argc > 7){
     shouldDumpClassConf = true;
     classConfs = (float*) malloc(sizeof(float) * test_input_size);
     predictedLabels = (int*) malloc(sizeof(int) * test_input_size);