我正在开发我的 MEX 文件,用于使用 CUDA 进行稀疏矩阵计算。我正在使用 CUSP 库。我不知道如何将 cusp::csr_matrix 返回到 Matlab。例如,我有
cusp::csr_matrix<mwIndex,double,cusp::host_memory> At(m,n,N);
因此,它是 CSR 格式的 At 矩阵,可以说,我已经计算过了。现在,我必须将它返回给 Matlab。像这样的东西:
plhs[0] = At;
但是,当然,它不是那样工作的,首先是因为 At 在 GPU 上。我想我应该使用 At.values 和索引方法。但是,如何将它们分配给主机数据?
有人可以建议如何做这一切吗?:)