From f8dd44d1806fb19418551eb6a97e28d7c0690d1a Mon Sep 17 00:00:00 2001
From: Blaise Tine <tinebp@yahoo.com>
Date: Thu, 17 Jun 2021 21:37:41 -0700
Subject: [PATCH] minor update

---
 tests/regression/io_addr/main.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/regression/io_addr/main.cpp b/tests/regression/io_addr/main.cpp
index 4c43838c..7899aa2a 100644
--- a/tests/regression/io_addr/main.cpp
+++ b/tests/regression/io_addr/main.cpp
@@ -183,9 +183,10 @@ int main(int argc, char *argv[]) {
   
   // allocate shared memory  
   std::cout << "allocate shared memory" << std::endl;    
-  uint32_t staging_buf_size = std::max<uint32_t>(src_buf_size,
+  uint32_t staging_buf_size = std::max<uint32_t>(NUM_ADDRS * sizeof(uint32_t),
+                                std::max<uint32_t>(src_buf_size,
                                   std::max<uint32_t>(dst_buf_size, 
-                                    sizeof(kernel_arg_t)));
+                                    sizeof(kernel_arg_t))));
   RT_CHECK(vx_alloc_shared_mem(device, staging_buf_size, &staging_buf));
   
   // upload kernel argument