mirror of
https://github.com/CNugteren/CLBlast.git
synced 2024-08-20 03:54:53 +02:00
86 lines
3 KiB
C++
86 lines
3 KiB
C++
|
|
// =================================================================================================
|
|
// This file is part of the CLBlast project. The project is licensed under the MIT license. This
|
|
// project loosely follows the Google C++ styleguide and uses a tab-size of two spaces and a max-
|
|
// width of 100 characters per line.
|
|
//
|
|
// Author(s):
|
|
// Cedric Nugteren <www.cedricnugteren.nl>
|
|
//
|
|
// This file tests any mat-vec-vec (A,X,Y) routine. It contains two types of tests: one testing
|
|
// all sorts of input combinations, and one deliberatly testing with invalid values.
|
|
//
|
|
// =================================================================================================
|
|
|
|
#ifndef CLBLAST_TEST_CORRECTNESS_TESTAXY_H_
|
|
#define CLBLAST_TEST_CORRECTNESS_TESTAXY_H_
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
#include "correctness/tester.h"
|
|
|
|
namespace clblast {
|
|
// =================================================================================================
|
|
|
|
// See comment at top of file for a description of the class
|
|
template <typename T>
|
|
class TestAXY: public Tester<T> {
|
|
public:
|
|
|
|
// Uses several variables from the Tester class
|
|
using Tester<T>::context_;
|
|
using Tester<T>::queue_;
|
|
using Tester<T>::kErrorMargin;
|
|
|
|
// Uses several helper functions from the Tester class
|
|
using Tester<T>::TestStart;
|
|
using Tester<T>::TestEnd;
|
|
using Tester<T>::TestSimilarity;
|
|
using Tester<T>::TestErrorCount;
|
|
using Tester<T>::TestErrorCodes;
|
|
using Tester<T>::GetExampleScalars;
|
|
|
|
// Test settings for the regular test. Append to this list in case more tests are required.
|
|
const std::vector<size_t> kMatrixVectorDims = { 61, 512 };
|
|
const std::vector<size_t> kOffsets = { 0, 10 };
|
|
const std::vector<size_t> kIncrements = { 1, 2 };
|
|
const std::vector<T> kAlphaValues = GetExampleScalars();
|
|
const std::vector<T> kBetaValues = GetExampleScalars();
|
|
|
|
// Test settings for the invalid test
|
|
const std::vector<size_t> kInvalidIncrements = { 0, 1 };
|
|
const size_t kBufferSize = 64;
|
|
|
|
// Shorthand for a BLAS routine
|
|
using Routine = std::function<StatusCode(const Arguments<T>&,
|
|
const Buffer&, const Buffer&, const Buffer&,
|
|
CommandQueue&)>;
|
|
|
|
// Constructor, initializes the base class tester and input data
|
|
TestAXY(const size_t platform_id, const size_t device_id,
|
|
const std::string &name, const std::vector<std::string> &options,
|
|
const Routine clblast_lambda, const Routine clblas_lambda);
|
|
|
|
// The test functions, taking no inputs
|
|
void TestRegular(Arguments<T> &args, const std::string &name);
|
|
void TestInvalidBufferSizes(Arguments<T> &args, const std::string &name);
|
|
|
|
private:
|
|
|
|
// Source data to test with
|
|
std::vector<T> a_source_;
|
|
std::vector<T> x_source_;
|
|
std::vector<T> y_source_;
|
|
|
|
// The routines to test
|
|
Routine clblast_lambda_;
|
|
Routine clblas_lambda_;
|
|
};
|
|
|
|
// =================================================================================================
|
|
} // namespace clblast
|
|
|
|
// CLBLAST_TEST_CORRECTNESS_TESTAXY_H_
|
|
#endif
|