From d22b2427e5fc743d84a75a256bc5dd63422ca831 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Mon, 22 Mar 2021 22:06:45 -0500
Subject: [PATCH] Fixing input size to 5K in cudnn HPVM src
---
.../benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp b/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp
index 569793db2a..9c0c980d97 100644
--- a/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp
+++ b/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp
@@ -491,7 +491,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;
std::string input_path = dir_prefix + std::string("tune_input.bin");
--
GitLab