diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-09-07 22:04:24 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-09-07 22:04:24 +0200 |
commit | b7d833901213d03fe5e7f10c15741f55c6c1eb54 (patch) | |
tree | 1dcd9a4ac6fc917fb44dd31fbe4d22a1fe71e8bf /test/correctness | |
parent | bbb4523b7cc664ad64cc17f7381e6bbfb0874f06 (diff) |
Reduced size of the xCONVGEMM correctness tests
Diffstat (limited to 'test/correctness')
-rw-r--r-- | test/correctness/testblas.hpp | 4 | ||||
-rw-r--r-- | test/correctness/tester.cpp | 1 |
2 files changed, 3 insertions, 2 deletions
diff --git a/test/correctness/testblas.hpp b/test/correctness/testblas.hpp index 6c0abab6..e9a995fd 100644 --- a/test/correctness/testblas.hpp +++ b/test/correctness/testblas.hpp @@ -139,9 +139,9 @@ template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kBatc template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kPadSizes = { 0, 1 }; template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kDilationSizes = { 1, 2 }; template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kKernelSizes = { 1, 3 }; -template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kNumKernels = { 1, 67 }; +template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kNumKernels = { 1, 6 }; template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kStrideValues = { 1, 3 }; -template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kChannelValues = { 1, 4 }; +template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kChannelValues = { 1, 2 }; // Test settings for the invalid tests template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kInvalidIncrements = { 0, 1 }; diff --git a/test/correctness/tester.cpp b/test/correctness/tester.cpp index d6a346a6..daa43f26 100644 --- a/test/correctness/tester.cpp +++ b/test/correctness/tester.cpp @@ -370,6 +370,7 @@ std::string Tester<T,U>::GetOptionsString(const Arguments<U> &args) { 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 == kArgNumKernels){result += kArgNumKernels + equals + ToString(args.num_kernels) + " "; } if (o == kArgKernelH) { result += kArgKernelH + equals + ToString(args.kernel_h) + " "; } if (o == kArgKernelW) { result += kArgKernelW + equals + ToString(args.kernel_w) + " "; } if (o == kArgPadH) { result += kArgPadH + equals + ToString(args.pad_h) + " "; } |