CUSP稀疏库给出了奇怪的结果

时间:2014-05-12 01:05:28

标签: c++ cuda cusp-library

我试图在CUSP中执行以下操作:

A=[
    1,1,0,0;
    2,2,2,0;
    0,3,3,3;
    0,0,4,4];
B=[1,1,1,1]';
disp(mldivide(A,B));

X=[0.9167,0.0833,-0.5000,0.7500]

另一方面,我从CUSP得到一个奇怪的答案

#include <cusp/dia_matrix.h>
#include <cusp/krylov/cg.h>
#include <cusp/print.h>
int main()
{
    cusp::dia_matrix<int,float,cusp::host_memory> A(4,4,10,3);
    A.diagonal_offsets[0] = -1;
    A.diagonal_offsets[1] =  0;
    A.diagonal_offsets[2] =  1;
    for (int i = 0;i <3;i++)
    {
        for (int q = 0 ;q < A.num_cols;q++)
        {
            A.values(q,i)=q+1;
        }
    }
    //copy
    cusp::dia_matrix<int,float,cusp::device_memory> AA = A;
    cusp::array1d<float,cusp::device_memory> BB(A.num_rows,1);
    cusp::array1d<float,cusp::device_memory> XX(A.num_rows,0);
    cusp::print(AA);
    cusp::print(XX);
    cusp::print(BB);
    cusp::krylov::cg(AA,XX,BB);\
    cusp::print(XX);
    return 0;
}

结果如

sparse matrix <4, 4> with 10 entries
              0              0              1
              0              1              1
              1              0              2
              1              1              2
              1              2              2
              2              1              3
              2              2              3
              2              3              3
              3              2              4
              3              3              4
array1d <4>
             0
             0
             0
             0
array1d <4>
             1
             1
             1
             1
array1d <4>
      -39.9938
       -53.436
       87.9025
      -30.1429

最后一个看起来不太合适。谁知道我做错了什么?我使用的代码是错误的还是我们应该有一个非常好的猜测解决方案+使用预处理器?

1 个答案:

答案 0 :(得分:2)

conjugate gradient method仅适用于对称正定矩阵。你的矩阵不对称。这就是为什么它不能(而且不能)产生有效的解决方案。使用适当的,良好调节的SPD矩阵,或使用不同的数值方法。