First step towards supporting im2col in the test infrastructure

1 files changed, 5 insertions, 0 deletions

diff --git a/test/correctness/tester.cpp b/test/correctness/tester.cpp
index fb85e609..1581fbfb 100644
--- a/test/correctness/tester.cpp
+++ b/test/correctness/tester.cpp
@@ -366,6 +366,11 @@ std::string Tester<T,U>::GetOptionsString(const Arguments<U> &args) {
     if (o == kArgAlpha)    { result += kArgAlpha + equals + ToString(args.alpha) + " "; }
     if (o == kArgBeta)     { result += kArgBeta + equals + ToString(args.beta) + " "; }
     if (o == kArgBatchCount){result += kArgBatchCount + equals + ToString(args.batch_count) + " "; }
+    if (o == kArgChannels) { result += kArgChannels + equals + ToString(args.channels) + " "; }
+    if (o == kArgHeight)   { result += kArgHeight + equals + ToString(args.height) + " "; }
+    if (o == kArgWidth)    { result += kArgWidth + equals + ToString(args.width) + " "; }
+    if (o == kArgKernelH)  { result += kArgKernelH + equals + ToString(args.kernel_h) + " "; }
+    if (o == kArgKernelW)  { result += kArgKernelW + equals + ToString(args.kernel_w) + " "; }
   }
   return result;
 }