aoclsparse_?mv() - 5.2 English - 68552

  1/* ************************************************************************
  2 * Copyright (c) 2023 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 <math.h>
 27#include <stdio.h>
 28
 29#define M 5
 30#define N 5
 31#define NNZ 8
 32
 33int main(void)
 34{
 35    aoclsparse_status     status;
 36    aoclsparse_matrix     A     = NULL;
 37    aoclsparse_mat_descr  descr = NULL;
 38    aoclsparse_operation  trans = aoclsparse_operation_none;
 39    aoclsparse_index_base base  = aoclsparse_index_base_zero;
 40
 41    double alpha = 1.0;
 42    double beta  = 0.0;
 43
 44    // Input matrix
 45    //  1  0  0  2  0
 46    //  0  3  0  0  0
 47    //  0  0  4  0  0
 48    //  0  5  0  6  7
 49    //  0  0  0  0  8
 50    aoclsparse_int csr_row_ptr[M + 1] = {0, 2, 3, 4, 7, 8};
 51    aoclsparse_int csr_col_ind[NNZ]   = {0, 3, 1, 2, 1, 3, 4, 4};
 52    double         csr_val[NNZ]       = {1, 2, 3, 4, 5, 6, 7, 8};
 53
 54    // Input vectors
 55    double x[N] = {1.0, 2.0, 3.0, 4.0, 5.0};
 56    double y[M];
 57
 58    // Expected reference result & tolerance
 59    double         y_exp[M] = {9.0, 6.0, 12.0, 69.0, 40.0};
 60    double         tol      = 1e-12;
 61    aoclsparse_int oki, ok = 1;
 62
 63    // Print aoclsparse version
 64    printf("%s\n", aoclsparse_get_version());
 65
 66    // Create matrix descriptor
 67    // aoclsparse_create_mat_descr sets aoclsparse_matrix_type to aoclsparse_matrix_type_general
 68    // and aoclsparse_index_base to aoclsparse_index_base_zero.
 69    aoclsparse_create_mat_descr(&descr);
 70
 71    // Initialise sparse matrix
 72    status = aoclsparse_create_dcsr(&A, base, M, N, NNZ, csr_row_ptr, csr_col_ind, csr_val);
 73    if(status != aoclsparse_status_success)
 74    {
 75        printf("Error while creating a sparse matrix, status = %i\n", status);
 76        return 1;
 77    }
 78
 79    // Hint the system what operation to expect // to identify hint id(which routine is to be executed, destroyed later)
 80    status = aoclsparse_set_mv_hint(A, trans, descr, 1);
 81    if(status != aoclsparse_status_success)
 82    {
 83        printf("Error while hinting operation, status = %i\n", status);
 84        return 1;
 85    }
 86
 87    // Optimize the matrix
 88    status = aoclsparse_optimize(A);
 89    if(status != aoclsparse_status_success)
 90    {
 91        printf("Error while optimizing the matrix, status = %i\n", status);
 92        return 1;
 93    }
 94
 95    // Invoke SPMV API (double precision)
 96    printf("Invoking aoclsparse_dmv...\n");
 97    status = aoclsparse_dmv(trans, &alpha, A, descr, x, &beta, y);
 98    if(status != aoclsparse_status_success)
 99    {
100        printf("Error while computing SPMV, status = %i\n", status);
101        return 1;
102    }
103
104    // Print and check the results
105    printf("Output vector:\n");
106    for(aoclsparse_int i = 0; i < M; i++)
107    {
108        oki = fabs(y[i] - y_exp[i]) <= tol;
109        printf("  %lf %c\n", y[i], oki ? ' ' : '!');
110        ok = ok && oki;
111    }
112
113    aoclsparse_destroy_mat_descr(descr);
114    aoclsparse_destroy(&A);
115    return ok ? 0 : 2;
116}