From 19924fac0093dcfc623c9cbd0ffda891dc4ce434 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Tue, 25 Feb 2020 17:39:59 -0600
Subject: [PATCH] Adding batch size as param to Autotuner DNN sources
---
.../dnn_sources/src/promise/alexnet2_promise.cc | 7 ++++++-
.../dnn_sources/src/promise/alexnet_promise.cc | 6 +++++-
.../dnn_sources/src/promise/lenet_promise.cc | 5 ++++-
.../dnn_sources/src/promise/mobilenet_promise.cc | 7 ++++++-
.../dnn_sources/src/promise/mobilenet_shallow_promise.cc | 7 ++++++-
.../dnn_sources/src/promise/resnet18_promise.cc | 6 +++++-
.../dnn_sources/src/promise/vgg16_cifar100_promise.cc | 8 ++++++--
.../dnn_sources/src/promise/vgg16_cifar10_promise.cc | 6 +++++-
8 files changed, 43 insertions(+), 9 deletions(-)
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 2cd364bf6b..a7333abd99 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 81cf894f43..3026f5e573 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 ff1fcba79c..41451598c5 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 691c97a3ef..47c1d01af9 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 89b3697298..120323e607 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 21c7b051c2..261753c1a9 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 27bc38e0e3..0918bb6855 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 c66de402d8..2fb5193ce0 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);
--
GitLab