diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/mobilenet/src/mobilenet_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/mobilenet/src/mobilenet_loop.cpp
index e22db1462ae0d7e27fbedc3f46a62556a57fcb80..096dcbce0b264a8deaa85e1225ed684172c29ab5 100644
--- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/mobilenet/src/mobilenet_loop.cpp
+++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/mobilenet/src/mobilenet_loop.cpp
@@ -2390,7 +2390,7 @@ int main(){
   args->dense_1_b_bytes = 0; 
 
   int batch_size = 500;
-  int test_input_size = 50000;  
+  int test_input_size = 5000;
   int batch_count = test_input_size / batch_size;
   
   // void* input = create4DTensor(0,nchw,batch_size,3,32,32);