Reduced size of the xCONVGEMM correctness tests
parent
bbb4523b7c
commit
b7d8339012
|
@ -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 };
|
||||
|
|
|
@ -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) + " "; }
|
||||
|
|
Loading…
Reference in New Issue