From e0c5312abb6f5b1c0e413fe69a8c7c46215cd3ae Mon Sep 17 00:00:00 2001 From: CNugteren Date: Mon, 14 Sep 2015 12:28:50 +0200 Subject: Added support for the dot buffer and offset argument --- test/correctness/testblas.h | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'test/correctness/testblas.h') diff --git a/test/correctness/testblas.h b/test/correctness/testblas.h index 8a86c65e..9e1d110c 100644 --- a/test/correctness/testblas.h +++ b/test/correctness/testblas.h @@ -90,6 +90,7 @@ class TestBlas: public Tester { std::vector a_source_; std::vector b_source_; std::vector c_source_; + std::vector dot_source_; // The routine-specific functions passed to the tester Routine run_routine_; @@ -136,6 +137,7 @@ void RunTests(int argc, char *argv[], const bool silent, const std::string &name auto a_offsets = std::vector{args.a_offset}; auto b_offsets = std::vector{args.b_offset}; auto c_offsets = std::vector{args.c_offset}; + auto dot_offsets = std::vector{args.dot_offset}; auto alphas = std::vector{args.alpha}; auto betas = std::vector{args.beta}; auto x_sizes = std::vector{args.x_size}; @@ -170,6 +172,7 @@ void RunTests(int argc, char *argv[], const bool silent, const std::string &name if (option == kArgAOffset) { a_offsets = tester.kOffsets; } if (option == kArgBOffset) { b_offsets = tester.kOffsets; } if (option == kArgCOffset) { c_offsets = tester.kOffsets; } + if (option == kArgDotOffset) { dot_offsets = tester.kOffsets; } if (option == kArgAlpha) { alphas = tester.kAlphaValues; } if (option == kArgBeta) { betas = tester.kBetaValues; } @@ -204,10 +207,12 @@ void RunTests(int argc, char *argv[], const bool silent, const std::string &name for (auto &b_offset: b_offsets) { r_args.b_offset = b_offset; for (auto &c_ld: c_lds) { r_args.c_ld = c_ld; for (auto &c_offset: c_offsets) { r_args.c_offset = c_offset; - for (auto &alpha: alphas) { r_args.alpha = alpha; - for (auto &beta: betas) { r_args.beta = beta; - C::SetSizes(r_args); - regular_test_vector.push_back(r_args); + for (auto &dot_offset: dot_offsets) { r_args.dot_offset = dot_offset; + for (auto &alpha: alphas) { r_args.alpha = alpha; + for (auto &beta: betas) { r_args.beta = beta; + C::SetSizes(r_args); + regular_test_vector.push_back(r_args); + } } } } -- cgit v1.2.3