aoclsparse_?dotmv() - 5.2 English - 68552

 1/* ************************************************************************
 2 * Copyright (c) 2023-2024 Advanced Micro Devices, Inc.
 3 *
 4 * Permission is hereby granted, free of charge, to any person obtaining a copy
 5 * of this software and associated documentation files (the "Software"), to deal
 6 * in the Software without restriction, including without limitation the rights
 7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 8 * copies of the Software, and to permit persons to whom the Software is
 9 * furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
20 * THE SOFTWARE.
21 *
22 * ************************************************************************ */
23
24#include "aoclsparse.h"
25
26#include <iostream>
27#include <limits>
28
29#define M 6
30#define N 5
31#define NNZ 8
32
33int main(void)
34{
35    aoclsparse_operation op = aoclsparse_operation_none;
36
37    double alpha = 1.0;
38    double beta  = 0.0;
39
40    // Print aoclsparse version
41    std::cout << aoclsparse_get_version() << std::endl;
42
43    // Create matrix descriptor
44    aoclsparse_mat_descr descr;
45    // aoclsparse_create_mat_descr set aoclsparse_matrix_type to aoclsparse_matrix_type_general
46    // and aoclsparse_index_base to aoclsparse_index_base_zero.
47    aoclsparse_create_mat_descr(&descr);
48
49    aoclsparse_index_base base = aoclsparse_index_base_zero;
50
51    // Initialise matrix
52    //  1  0  0  2  0
53    //  0  3  0  0  0
54    //  0  0  4  0  0
55    //  0  5  0  6  7
56    //  0  0  0  0  8
57    //  0  0  0  0  0
58    aoclsparse_int    csr_row_ptr[M + 1] = {0, 2, 3, 4, 7, 8, 8};
59    aoclsparse_int    csr_col_ind[NNZ]   = {0, 3, 1, 2, 1, 3, 4, 4};
60    double            csr_val[NNZ]       = {1, 2, 3, 4, 5, 6, 7, 8};
61    aoclsparse_matrix A;
62    aoclsparse_create_dcsr(&A, base, M, N, NNZ, csr_row_ptr, csr_col_ind, csr_val);
63
64    // Initialise vectors
65    double x[N]     = {1.0, 1.0, 1.0, 1.0, 1.0};
66    double y[M]     = {1.0, 1.0, 1.0, 1.0, 1.0, 1.0};
67    double d        = 0;
68    double y_exp[M] = {3.0, 3.0, 4.0, 18.0, 8.0};
69    double d_exp    = 36;
70
71    //to identify hint id(which routine is to be executed, destroyed later)
72    aoclsparse_set_dotmv_hint(A, op, descr, 1);
73
74    // Optimize the matrix, "A"
75    aoclsparse_optimize(A);
76
77    std::cout << "Invoking aoclsparse_ddotmv..";
78    //Invoke DOTMV API (double precision)
79    aoclsparse_ddotmv(op, alpha, A, descr, x, beta, y, &d);
80    std::cout << "Done." << std::endl;
81
82    // Print and check the result
83    std::cout << std::endl << "y = ";
84    bool   oki, ok = true;
85    double tol = 20 * std::numeric_limits<double>::epsilon();
86    for(aoclsparse_int i = 0; i < M; i++)
87    {
88        oki = std::abs(y[i] - y_exp[i]) <= tol;
89        std::cout << y[i] << (oki ? " " : "!  ");
90        ok &= oki;
91    }
92
93    ok &= (std::abs(d - d_exp) <= tol);
94    std::cout << std::endl << "Output dot product: " << d << (ok ? " " : "!  ") << std::endl;
95
96    aoclsparse_destroy_mat_descr(descr);
97    aoclsparse_destroy(&A);
98    return ok ? 0 : 4;
99}