summaryrefslogtreecommitdiff
path: root/external/clBLAS/src/tests/include/trmv.h
blob: 204bbce02bbbdf3fa35a3f40fefefa499e2c6a11 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/* ************************************************************************
 * Copyright 2013 Advanced Micro Devices, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ************************************************************************/


#if !defined(TRMV_PACKED)
    #ifndef TRMV_H
        #define TRMV_H
    #else
        #define DUPLICIT
    #endif
#endif

#ifndef DUPLICIT

#include <gtest/gtest.h>
#include <clBLAS.h>
#include <common.h>
#include <BlasBase.h>
#include <blas-math.h>

using ::testing::TestWithParam;

#ifndef TRMV_PACKED
class TRMV : public TestWithParam<
#else
class TPMV : public TestWithParam<
#endif

    ::std::tr1::tuple<
    clblasOrder,     // order
	clblasUplo,		// uplo
    clblasTranspose, // transA
	clblasDiag,		// diag
    int,                // N
    int,                // lda, 0 - undefined
    int,                // incx, should be greater than 0
	int,				//offa
	int,				//offx
    int                 // numCommandQueues
        > > {
public:
    void getParams(TestParams *params)
    {
        params->order = order;
		params->uplo = uplo;
        params->transA = transA;
		params->diag = diag;
        params->seed = seed;
        params->N = N;
        params->lda = lda;
        params->incx = incx;
		params->offa = offa;
		params->offBX = offx;
        params->numCommandQueues = numCommandQueues;
    }

protected:
    virtual void SetUp()
    {
        order = ::std::tr1::get<0>(GetParam());
        uplo = ::std::tr1::get<1>(GetParam());
		transA = ::std::tr1::get<2>(GetParam());
		diag = ::std::tr1::get<3>(GetParam());
        N = ::std::tr1::get<4>(GetParam());
        lda = ::std::tr1::get<5>(GetParam());
        incx = ::std::tr1::get<6>(GetParam());
		offa = ::std::tr1::get<7>(GetParam());
		offx = ::std::tr1::get<8>(GetParam());
        numCommandQueues = ::std::tr1::get<9>(GetParam());

        #ifndef TRMV_PACKED
		lda = ::std::max( lda, N );
        #else
        lda =0;
        #endif

        base = ::clMath::BlasBase::getInstance();
        seed = base->seed();

        useNumCommandQueues = base->useNumCommandQueues();
        if (useNumCommandQueues) {
            numCommandQueues = base->numCommandQueues();
        }

        if (base->useN()) {
            N = base->N();
        }

		printTestParams(order, uplo, transA, diag, N, lda, incx, offa, offx);
			::std::cerr << "seed = " << seed << ::std::endl;
			::std::cerr << "queues = " << numCommandQueues << ::std::endl;
    }

    clblasOrder order;
	clblasUplo uplo;
    clblasTranspose transA;
	clblasDiag diag;
    size_t N;
    size_t lda;
    int incx;
    size_t offx, offa;
    unsigned int seed;

    ::clMath::BlasBase *base;
    cl_ulong imageA, imageX;

    bool useNumCommandQueues;
    cl_uint numCommandQueues;
};

#endif