diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-21 11:47:16 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-21 11:47:16 +0200 |
commit | 838422fbb1a8fa7ce2cad06bb94b2779d3929e08 (patch) | |
tree | 922bc405046de69cc6d9f0630db465be96332f31 /test/correctness | |
parent | 5d87abf78080de8e844ff93822da49d2c8a7deb3 (diff) |
Further implemented single-kernel approach of convgemm; extended test to capture other parts of the kernel code
Diffstat (limited to 'test/correctness')
-rw-r--r-- | test/correctness/testblas.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/correctness/testblas.hpp b/test/correctness/testblas.hpp index 0dc8584e..6c0abab6 100644 --- a/test/correctness/testblas.hpp +++ b/test/correctness/testblas.hpp @@ -139,7 +139,7 @@ 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, 2 }; +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>::kStrideValues = { 1, 3 }; template <typename T, typename U> const std::vector<size_t> TestBlas<T,U>::kChannelValues = { 1, 4 }; |