diff options
Diffstat (limited to 'test/correctness/testblas.hpp')
-rw-r--r-- | test/correctness/testblas.hpp | 4 |
1 files changed, 2 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 }; |