-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;
}

但我得到的结果是 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

有什么帮助吗？

谢谢。

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 回答