From 7f496c1bb827b0e8a05bef8f92207185d05173db Mon Sep 17 00:00:00 2001
From: Maria Kotsifakou <kotsifa2@illinois.edu>
Date: Fri, 15 Nov 2019 17:52:23 -0600
Subject: [PATCH] Updated HPVM loop sources (input data sizes)

---
 .../DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp    | 2 +-
 .../DNN_Benchmarks/benchmarks/alexnet2/src/alexnet2_loop.cpp  | 2 +-
 .../DNN_Benchmarks/benchmarks/lenet_mnist/src/lenet_loop.cpp  | 4 ++--
 .../DNN_Benchmarks/benchmarks/resnet18/src/resnet18_loop.cpp  | 2 +-
 .../benchmarks/vgg16_cifar10/src/vgg16_cifar10_loop.cpp       | 2 +-
 .../benchmarks/vgg16_cifar100/src/vgg16_cifar100_loop.cpp     | 2 +-
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp
index 50732550db..a62b2ea33a 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet/src/alexnet_loop.cpp
@@ -437,7 +437,7 @@ int main(){
   args->dense_1_b_bytes = 0; 
 
   int batch_size = 500; 
-  int test_input_size = 10000;
+  int test_input_size = 5000;
   int batch_count = test_input_size / batch_size; 
   
   // void* input = create4DTensor(0,nchw,batch_size,3,32,32);
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet2/src/alexnet2_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet2/src/alexnet2_loop.cpp
index 91bf3b0c45..e3c06325a6 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet2/src/alexnet2_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/alexnet2/src/alexnet2_loop.cpp
@@ -487,7 +487,7 @@ int main(){
 
 
   int batch_size = 500;
-  int test_input_size = 10000;  
+  int test_input_size = 5000;  
   int batch_count = test_input_size / batch_size;
 
   std::string input_path =  dir_prefix + std::string("input.bin"); 
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/lenet_mnist/src/lenet_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/lenet_mnist/src/lenet_loop.cpp
index f3b20f0112..88f3022569 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/lenet_mnist/src/lenet_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/lenet_mnist/src/lenet_loop.cpp
@@ -321,7 +321,7 @@ int main(){
 
 
   int batch_size = 500;
-  int test_input_size = 10000;
+  int test_input_size = 5000;
   int batch_count = test_input_size / batch_size;
 
   startMemTracking();
@@ -334,7 +334,7 @@ int main(){
 
     void* input = readInputBatch(input_path.c_str(), 0,
                                  start, end, 
-                                 3, 32, 32);
+                                 1, 28, 28);
      
     args->input = input;
     args->input_bytes = 0;
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/resnet18/src/resnet18_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/resnet18/src/resnet18_loop.cpp
index 1f81dba0c7..4c91f0e661 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/resnet18/src/resnet18_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/resnet18/src/resnet18_loop.cpp
@@ -1427,7 +1427,7 @@ int main(){
   args->dense_1_b_bytes = 0; 
 
   int batch_size = 500;
-  int test_input_size = 10000;
+  int test_input_size = 5000;
   int batch_count = test_input_size / batch_size;
 
   // NOTE-HASHIM: commented out
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar10/src/vgg16_cifar10_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar10/src/vgg16_cifar10_loop.cpp
index a044290425..793139b9a8 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar10/src/vgg16_cifar10_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar10/src/vgg16_cifar10_loop.cpp
@@ -971,7 +971,7 @@ int main(){
 
 
   int batch_size = 500;
-  int test_input_size = 10000;
+  int test_input_size = 5000;
   int batch_count = test_input_size / batch_size;
 
   // void* input = create4DTensor(0,nchw,batch_size,3,32,32);
diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar100/src/vgg16_cifar100_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar100/src/vgg16_cifar100_loop.cpp
index b8126be314..7bb78e07bb 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar100/src/vgg16_cifar100_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_cifar100/src/vgg16_cifar100_loop.cpp
@@ -973,7 +973,7 @@ int main(){
 
 
   int batch_size = 500;
-  int test_input_size = 10000;
+  int test_input_size = 5000;
   int batch_count = test_input_size / batch_size;
 
   startMemTracking();
-- 
GitLab