-1

CUSP と CUDA を使用して疎行列を csr 形式に変換しようとしています。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;
}

しかし、得られる結果は、インデックスが 1 シフトされた coo 形式の行列です。

%%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

助けはありますか？

ありがとう。

4

1 に答える 1

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";
于 2015-06-10T04:12:32.023 に答える