Adjusted the test-infrastructure to support testing of batched-versions of routines

1 files changed, 3 insertions, 0 deletions

diff --git a/src/utilities/utilities.hpp b/src/utilities/utilities.hpp
index 330db597..58dc3b27 100644
--- a/src/utilities/utilities.hpp
+++ b/src/utilities/utilities.hpp
@@ -73,6 +73,7 @@ constexpr auto kArgAsumOffset = "offasum";
 constexpr auto kArgImaxOffset = "offimax";
 constexpr auto kArgAlpha = "alpha";
 constexpr auto kArgBeta = "beta";
+constexpr auto kArgBatchCount = "batch_count";
 
 // The tuner-specific arguments in string form
 constexpr auto kArgFraction = "fraction";
@@ -156,6 +157,8 @@ struct Arguments {
   size_t imax_offset = 0;
   T alpha = ConstantOne<T>();
   T beta = ConstantOne<T>();
+  size_t batch_count = 1;
+  // Sizes
   size_t x_size = 1;
   size_t y_size = 1;
   size_t a_size = 1;