@@ -49,10 +49,12 @@ PROGRAM TEST_SPBLAS
49
49
! | 0 0 50 60 70 0 |
50
50
! | 0 0 0 0 0 80 |
51
51
52
- ! create csr
52
+ x = [1 , 1 , 1 , 1 , 1 , 1 ] ! for spmv
53
+ values = [10 , 20 , 30 , 40 , 50 , 60 , 70 , 80 ] ! for matrix values
54
+
55
+ ! ! create csr
53
56
ia = [1 , 3 , 5 , 8 , 9 ]
54
57
ja = [1 , 2 , 2 , 4 , 3 , 4 , 5 , 6 ]
55
- values = [10 , 20 , 30 , 40 , 50 , 60 , 70 , 80 ]
56
58
57
59
stat = mkl_sparse_s_create_csr( &
58
60
a, SPARSE_INDEX_BASE_ONE, rows, cols, ia(1 :4 ), ia(2 :5 ), ja, values)
@@ -61,22 +63,24 @@ PROGRAM TEST_SPBLAS
61
63
descr% TYPE = SPARSE_MATRIX_TYPE_GENERAL
62
64
63
65
! create coo
64
- row_indx = [1 , 1 , 2 , 2 , 3 , 3 , 3 , 4 ]
65
- col_indx = [1 , 2 , 2 , 4 , 3 , 4 , 5 , 6 ]
66
- stat = mkl_sparse_s_create_coo( &
67
- A_coo, SPARSE_INDEX_BASE_ONE, rows, cols, nnz, row_indx, col_indx, &
68
- values)
66
+ ! row_indx = [1, 1, 2, 2, 3, 3, 3, 4]
67
+ ! col_indx = [1, 2, 2, 4, 3, 4, 5, 6]
68
+ ! stat = mkl_sparse_s_create_coo( &
69
+ ! A_coo, SPARSE_INDEX_BASE_ONE, rows, cols, nnz, row_indx, col_indx, &
70
+ ! values)
71
+ ! PRINT *, "stat create = ", stat
72
+ ! descr%TYPE = SPARSE_MATRIX_TYPE_GENERAL
73
+ ! PRINT *, "descr%TYPE", descr%TYPE
69
74
70
75
! spmv csr
71
- x = [1 , 1 , 1 , 1 , 1 , 1 ]
72
76
stat = mkl_sparse_s_mv(SPARSE_OPERATION_NON_TRANSPOSE, 1.0 , a, descr, x, 0.0 , y)
73
77
PRINT * , " stat mv = " , stat
74
78
PRINT * , " result csr = " , y
75
79
PRINT * , " expected = " , [30 ., 70 ., 180 ., 80 .]
76
80
77
- ! spmv coo
78
- stat = mkl_sparse_s_mv( &
79
- SPARSE_OPERATION_NON_TRANSPOSE, 1.0 , A_coo, descr, x, 0.0 , y_coo)
80
- PRINT * , " result coo = " , y_coo
81
+ ! ! spmv coo
82
+ ! stat = mkl_sparse_s_mv( &
83
+ ! SPARSE_OPERATION_NON_TRANSPOSE, 1.0, A_coo, descr, x, 0.0, y_coo)
84
+ ! PRINT *, "result coo = ", y_coo
81
85
82
86
END PROGRAM TEST_SPBLAS
0 commit comments