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
|
#include <cusp/hyb_matrix.h>
#include <cusp/print.h>
int main(void)
{
// allocate storage for (4,3) matrix with 8 nonzeros
// ELL portion has 5 nonzeros and storage for 2 nonzeros per row
// COO portion has 3 nonzeros
cusp::hyb_matrix<int, float, cusp::host_memory> A(3, 4, 5, 3, 2);
// Initialize A to represent the following matrix
// [10 20 30 40]
// [ 0 50 0 0]
// [60 0 70 80]
// A is split into ELL and COO parts as follows
// [10 20 30 40] [10 20 0 0] [ 0 0 30 40]
// [ 0 50 0 0] = [ 0 50 0 0] + [ 0 0 0 0]
// [60 0 70 80] [60 0 70 0] [ 0 0 0 80]
// Initialize ELL part
// X is used to fill unused entries in the ELL portion of the matrix
const int X = cusp::ell_matrix<int,float,cusp::host_memory>::invalid_index;
// first row
A.ell.column_indices(0,0) = 0; A.ell.values(0,0) = 10;
A.ell.column_indices(0,1) = 1; A.ell.values(0,1) = 20;
// second row
A.ell.column_indices(1,0) = 1; A.ell.values(1,0) = 50; // shifted to leftmost position
A.ell.column_indices(1,1) = X; A.ell.values(1,1) = 0; // padding
// third row
A.ell.column_indices(2,0) = 0; A.ell.values(2,0) = 60;
A.ell.column_indices(2,1) = 2; A.ell.values(2,1) = 70; // shifted to leftmost position
// Initialize COO part
A.coo.row_indices[0] = 0; A.coo.column_indices[0] = 2; A.coo.values[0] = 30;
A.coo.row_indices[1] = 0; A.coo.column_indices[1] = 3; A.coo.values[1] = 40;
A.coo.row_indices[2] = 2; A.coo.column_indices[2] = 3; A.coo.values[2] = 80;
// print matrix entries
cusp::print(A);
return 0;
}
|
