File: testing_const_dnvec_descr.cpp

package info (click to toggle)
rocsparse 6.4.3-3
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 20,428 kB
  • sloc: cpp: 234,069; f90: 9,307; sh: 2,262; python: 1,939; makefile: 1,585; ansic: 440; xml: 26
file content (122 lines) | stat: -rw-r--r-- 4,833 bytes parent folder | download
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
/* ************************************************************************
 * Copyright (C) 2023 Advanced Micro Devices, Inc. All rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 * ************************************************************************ */

#include "testing.hpp"

template <typename T>
void testing_const_dnvec_descr_bad_arg(const Arguments& arg)
{
    static const size_t safe_size = 100;

    rocsparse_const_dnvec_descr local_descr;
    int64_t                     local_size      = safe_size;
    rocsparse_datatype          local_data_type = get_datatype<T>();

    {
        rocsparse_const_dnvec_descr* descr     = &local_descr;
        int64_t                      size      = local_size;
        const void*                  values    = (const void*)0x4;
        rocsparse_datatype           data_type = local_data_type;

#define PARAMS_CREATE descr, size, values, data_type
        bad_arg_analysis(rocsparse_create_const_dnvec_descr, PARAMS_CREATE);
#undef PARAMS_CREATE
        // Check valid descriptor creations
        EXPECT_ROCSPARSE_STATUS(rocsparse_create_const_dnvec_descr(descr, 0, nullptr, data_type),
                                rocsparse_status_success);
        EXPECT_ROCSPARSE_STATUS(rocsparse_destroy_dnvec_descr(local_descr),
                                rocsparse_status_success);
        // rocsparse_destroy_dnvec_descr
        EXPECT_ROCSPARSE_STATUS(rocsparse_destroy_dnvec_descr(nullptr),
                                rocsparse_status_invalid_pointer);

        // Create valid descriptor
        EXPECT_ROCSPARSE_STATUS(rocsparse_create_const_dnvec_descr(descr, size, values, data_type),
                                rocsparse_status_success);
    }

    {
        rocsparse_const_dnvec_descr descr     = local_descr;
        int64_t*                    size      = &local_size;
        const void**                values    = (const void**)0x4;
        rocsparse_datatype*         data_type = &local_data_type;
#define PARAMS_GET descr, size, values, data_type
        bad_arg_analysis(rocsparse_const_dnvec_get, PARAMS_GET);
#undef PARAMS_GET
    }

    // Destroy valid descriptor
    EXPECT_ROCSPARSE_STATUS(rocsparse_destroy_dnvec_descr(local_descr), rocsparse_status_success);
}

template <typename T>
void testing_const_dnvec_descr(const Arguments& arg)
{
    int64_t            m     = arg.M;
    rocsparse_datatype ttype = get_datatype<T>();

    device_vector<T> values(m);

    if(arg.unit_check)
    {
        rocsparse_const_dnvec_descr A{};

        // Create valid descriptor
        CHECK_ROCSPARSE_ERROR(
            rocsparse_create_const_dnvec_descr(&A, m, (const void*)values, ttype));

        int64_t            gold_m;
        rocsparse_datatype gold_ttype;
        const T*           gold_values;

        CHECK_ROCSPARSE_ERROR(
            rocsparse_const_dnvec_get(A, &gold_m, (const void**)&gold_values, &gold_ttype));

        unit_check_scalar<int64_t>(m, gold_m);
        unit_check_enum<rocsparse_datatype>(ttype, gold_ttype);

        ASSERT_EQ(values, (const T*)gold_values);

        gold_values = nullptr;

        CHECK_ROCSPARSE_ERROR(rocsparse_const_dnvec_get_values(A, (const void**)&gold_values));

        ASSERT_EQ(values, (const T*)gold_values);

        CHECK_ROCSPARSE_ERROR(rocsparse_destroy_dnvec_descr(A));
    }

    if(arg.timing)
    {
    }
}

#define INSTANTIATE(TTYPE)                                                        \
    template void testing_const_dnvec_descr_bad_arg<TTYPE>(const Arguments& arg); \
    template void testing_const_dnvec_descr<TTYPE>(const Arguments& arg)

INSTANTIATE(float);
INSTANTIATE(double);
INSTANTIATE(rocsparse_float_complex);
INSTANTIATE(rocsparse_double_complex);
void testing_const_dnvec_descr_extra(const Arguments& arg) {}