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 50732550db8c8f02c940e485702c3253a7bb9760..a62b2ea33ad7cc2838e68965e2ce19f615555b93 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 91bf3b0c4523e7239d7f11f6ad350f9dbb454a91..e3c06325a67e648da47f710c72498308da1041be 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 f3b20f01121f667562a2ef72d99e3f518ee84af6..88f302256933bae34169fa051559ef42adfd321b 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 1f81dba0c7ca6d5334565b34d7081b6155fa0569..4c91f0e6619558d30f3819a6161fb73e5b6aaf7f 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 a0442904255ba6c6a6399205775693be1952df09..793139b9a868b2d7a80b1c7352d83056853a8175 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 b8126be3142749280791687b634dc6e697f07b77..7bb78e07bb8d6bda27b961753c9ccf3771151863 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();