使用cusp库乘以2个稀疏矩阵

Question

我刚开始使用cuda的cusp库。我正在尝试为CUDA实现修改后的单纯形算法。为此，我需要乘以2个稀疏矩阵来更新基础矩阵。

所以问题是-我如何使用cusp库将2个稀疏矩阵相乘(不管是什么格式)？另外，有没有办法知道结果矩阵将包含多少非零元素(用于内存分配目的)？

我试过了：

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdlib.h>
#include <stdio.h>
#include "cusparse.h"
#include <cusp/version.h>
#include <cusp/multiply.h>
#include <cusp/array2d.h>
#include <cusp/print.h>
#include <cusp/coo_matrix.h>

int main(void)
{
    cusp::coo_matrix<int,double,cusp::device_memory> A(2,2,2);

    A.values[0] = 1;
    A.row_indices[0] = 0;
    A.column_indices[0]= 0;

    A.values[1] = 1;
    A.row_indices[1] = 1;
    A.column_indices[1]= 1;

    cusp::coo_matrix<int, double, cusp::device_memory> B(2,2,4);

    B.values[0] = 1;
    B.row_indices[0] = 0;
    B.column_indices[0]= 0;

    B.values[1] = 2;
    B.row_indices[1] = 0;
    B.column_indices[1]= 1;

    B.values[2] = 3;
    B.row_indices[2] = 1;
    B.column_indices[2]= 0;

    B.values[3] = 4;
    B.row_indices[3] = 1;
    B.column_indices[3]= 1;

    cusp::print(A);
    cusp::print(B);

    cusp::coo_matrix<int,double, cusp::device_memory> C(2,2,4);

    cusp::multiply(A,B,C);

    cusp::print(C);

}

As cusp:multiply()是我发现的唯一用于乘法的函数。

cusp v.0.4 CUDA v.5.5

Answer 1

如何使用cusp库将2个稀疏矩阵相乘(不管是什么格式)？

是的，cusp::multiply是正确的函数。你的代码对我来说是正确的。

是否有办法知道结果矩阵将包含多少非零元素(用于内存分配目的)？

你不需要提前分配它。将C矩阵的定义更改为：

cusp::coo_matrix<int,double, cusp::device_memory> C;

并且您的代码仍然可以正常工作。

Answer 2

感谢您的回答。这个代码对我不起作用，但我找到了问题所在。由于某些原因，当我将计算能力设置为3.5，SM版本也设置为3.5时，它无法工作。我将其设置为2.0/2.1，现在它工作得很好。