Распечатать csr формат матрицы с CUSP

Question

Я пытаюсь преобразовать разреженную матрицу в формат CSR, используя CUSP и CUDA. Я немного изменил код, показанный в документации CUSP:

#include <cusp/array2d.h>
#include <cusp/coo_matrix.h>
#include <cusp/csr_matrix.h>
#include <cusp/print.h>
int main(void)
{
// create a simple example
cusp::array2d<float, cusp::host_memory> A(3,4);
A(0,0) = 10;  A(0,1) =  0;  A(0,2) = 20;  A(0,3) =  0;
A(1,0) =  0;  A(1,1) = 30;  A(1,2) =  0;  A(1,3) = 40;
A(2,0) = 50;  A(2,1) = 60;  A(2,2) = 70;  A(2,3) = 80;
print(A);
// save A to disk in MatrixMarket format
cusp::io::write_matrix_market_file(A, "A.mtx");
// load A from disk into a coo_matrix
cusp::csr_matrix<int, float, cusp::device_memory> B;
cusp::io::read_matrix_market_file(B, "A.mtx");

cusp::io::write_matrix_market_file(B,"B.mtx");
// print B
cusp::print(B);
return 0;
}

Но в результате я получаю матрицу в формате coo с индексами, сдвинутыми на 1:

%%MatrixMarket matrix coordinate real general
3   4   8
1 1 10
1 3 20
2 2 30
2 4 40
3 1 50
3 2 60
3 3 70
3 4 80

Любая помощь?

Благодарю.

Answer 1

Спасибо за ответы! На самом деле мне пришлось писать формат csr вручную, хотя это было не так сложно, как я изначально думал. Мой код выглядел так:

// Imprimimos la matriz en formato csr:
cout<<"\n=======================\n";
cout<<"Matriz de adyacencia en formato CSR:\n";
cout<<"Row-offset   :";
for (int j=0 ; j<10 ; j++) cout<<B.row_offsets[j]<<" ";
cout<<"\n"; cout<<"Column index :";
for (int j=0 ; j<17 ; j++) cout<<B.column_indices[j]<<" ";
cout<<"\n"; cout<<"Values:";
for (int j=0 ; j<17 ; j++) cout<<B.values[j]<<" ";   
cout<<"\n==================\n";