diff options
Diffstat (limited to 'test/correctness/routines/xgemm.cc')
-rw-r--r-- | test/correctness/routines/xgemm.cc | 124 |
1 files changed, 64 insertions, 60 deletions
diff --git a/test/correctness/routines/xgemm.cc b/test/correctness/routines/xgemm.cc index b829aaa3..2c0a3492 100644 --- a/test/correctness/routines/xgemm.cc +++ b/test/correctness/routines/xgemm.cc @@ -7,77 +7,81 @@ // Author(s): // Cedric Nugteren <www.cedricnugteren.nl> // -// This file implements the tests for the Xgemm routine. It is based on the TestABC class. +// This file implements the tests for the Xgemm routine. // // ================================================================================================= -#include "wrapper_clblas.h" -#include "correctness/testabc.h" +#include "correctness/testblas.h" +#include "routines/xgemm.h" namespace clblast { // ================================================================================================= -// The correctness tester, containing the function calls to CLBlast and to clBLAS for comparison. +// The correctness tester template <typename T> -void XgemmTest(int argc, char *argv[], const bool silent, const std::string &name) { +void RunTest(int argc, char *argv[], const bool silent, const std::string &name) { - // Creates the CLBlast lambda - auto clblast_lambda = [](const Arguments<T> &args, - const Buffer &a_mat, const Buffer &b_mat, const Buffer &c_mat, - CommandQueue &queue) -> StatusCode { - auto queue_plain = queue(); - auto event = cl_event{}; - return Gemm(args.layout, args.a_transpose, args.b_transpose, - args.m, args.n, args.k, - args.alpha, - a_mat(), args.a_offset, args.a_ld, - b_mat(), args.b_offset, args.b_ld, - args.beta, - c_mat(), args.c_offset, args.c_ld, - &queue_plain, &event); - }; - - // Creates the clBLAS lambda (for comparison) - auto clblas_lambda = [](const Arguments<T> &args, - const Buffer &a_mat, const Buffer &b_mat, const Buffer &c_mat, - CommandQueue &queue) -> StatusCode { - auto queue_plain = queue(); - auto event = cl_event{}; - auto status = clblasXgemm(static_cast<clblasOrder>(args.layout), - static_cast<clblasTranspose>(args.a_transpose), - static_cast<clblasTranspose>(args.b_transpose), - args.m, args.n, args.k, - args.alpha, - a_mat(), args.a_offset, args.a_ld, - b_mat(), args.b_offset, args.b_ld, - args.beta, - c_mat(), args.c_offset, args.c_ld, - 1, &queue_plain, 0, nullptr, &event); - return static_cast<StatusCode>(status); - }; + // Creates a tester + TestBlas<T> tester{argc, argv, silent, name, TestXgemm<T>::GetOptions(), + TestXgemm<T>::RunRoutine, TestXgemm<T>::RunReference, + TestXgemm<T>::DownloadResult, TestXgemm<T>::GetResultIndex, + TestXgemm<T>::ResultID1, TestXgemm<T>::ResultID2}; - // Initializes the arguments relevant for this routine + // This variable holds the arguments relevant for this routine auto args = Arguments<T>{}; - const auto options = std::vector<std::string>{kArgM, kArgN, kArgK, kArgLayout, - kArgATransp, kArgBTransp, - kArgALeadDim, kArgBLeadDim, kArgCLeadDim, - kArgAOffset, kArgBOffset, kArgCOffset}; - - // Creates a tester - TestABC<T> tester{argc, argv, silent, name, options, clblast_lambda, clblas_lambda}; // Loops over the test-cases from a data-layout point of view - for (auto &layout: tester.kLayouts) { - args.layout = layout; - for (auto &a_transpose: tester.kTransposes) { - args.a_transpose = a_transpose; - for (auto &b_transpose: tester.kTransposes) { - args.b_transpose = b_transpose; - const auto case_name = ToString(layout)+" "+ToString(a_transpose)+" "+ToString(b_transpose); + for (auto &layout: tester.kLayouts) { args.layout = layout; + for (auto &a_transpose: tester.kTransposes) { args.a_transpose = a_transpose; + for (auto &b_transpose: tester.kTransposes) { args.b_transpose = b_transpose; + + // Creates the arguments vector for the regular tests + auto regular_test_vector = std::vector<Arguments<T>>{}; + for (auto &m: tester.kMatrixDims) { args.m = m; + for (auto &n: tester.kMatrixDims) { args.n = n; + for (auto &k: tester.kMatrixDims) { args.k = k; + for (auto &a_ld: tester.kMatrixDims) { args.a_ld = a_ld; + for (auto &a_offset: tester.kOffsets) { args.a_offset = a_offset; + for (auto &b_ld: tester.kMatrixDims) { args.b_ld = b_ld; + for (auto &b_offset: tester.kOffsets) { args.b_offset = b_offset; + for (auto &c_ld: tester.kMatrixDims) { args.c_ld = c_ld; + for (auto &c_offset: tester.kOffsets) { args.c_offset = c_offset; + for (auto &alpha: tester.kAlphaValues) { args.alpha = alpha; + for (auto &beta: tester.kBetaValues) { args.beta = beta; + args.a_size = TestXgemm<T>::GetSizeA(args); + args.b_size = TestXgemm<T>::GetSizeB(args); + args.c_size = TestXgemm<T>::GetSizeC(args); + if (args.a_size<1 || args.b_size<1 || args.c_size<1) { continue; } + regular_test_vector.push_back(args); + } + } + } + } + } + } + } + } + } + } + } + + // Creates the arguments vector for the invalid-buffer tests + auto invalid_test_vector = std::vector<Arguments<T>>{}; + args.m = args.n = args.k = tester.kBufferSize; + args.a_ld = args.b_ld = args.c_ld = tester.kBufferSize; + args.a_offset = args.b_offset = args.c_offset = 0; + for (auto &a_size: tester.kMatSizes) { args.a_size = a_size; + for (auto &b_size: tester.kMatSizes) { args.b_size = b_size; + for (auto &c_size: tester.kMatSizes) { args.c_size = c_size; + invalid_test_vector.push_back(args); + } + } + } // Runs the tests - tester.TestRegular(args, case_name, false); - tester.TestInvalidBufferSizes(args, case_name); + const auto case_name = ToString(layout)+" "+ToString(a_transpose)+" "+ToString(b_transpose); + tester.TestRegular(regular_test_vector, case_name); + tester.TestInvalid(invalid_test_vector, case_name); } } } @@ -88,10 +92,10 @@ void XgemmTest(int argc, char *argv[], const bool silent, const std::string &nam // Main function (not within the clblast namespace) int main(int argc, char *argv[]) { - clblast::XgemmTest<float>(argc, argv, false, "SGEMM"); - clblast::XgemmTest<double>(argc, argv, true, "DGEMM"); - clblast::XgemmTest<clblast::float2>(argc, argv, true, "CGEMM"); - clblast::XgemmTest<clblast::double2>(argc, argv, true, "ZGEMM"); + clblast::RunTest<float>(argc, argv, false, "SGEMM"); + clblast::RunTest<double>(argc, argv, true, "DGEMM"); + clblast::RunTest<clblast::float2>(argc, argv, true, "CGEMM"); + clblast::RunTest<clblast::double2>(argc, argv, true, "ZGEMM"); return 0; } |