Question

我正在尝试使用CUSP作为Mathematica的外部线性求解器来使用GPU的强大功能。这是CUSP Project webpage。我想问一下如何将CUSP与Mathematica集成。我相信你们中的许多人都有兴趣讨论这个问题。我认为编写输入矩阵然后将其提供给CUSP程序是不可取的。使用Mathematica的LibrarayFunctionLoad将是一种更好的方法，可以将输入矩阵传输到基于GPU的求解器。直接从Mathematica提供矩阵和右侧矩阵的方法是什么？

以下是一些CUSP代码段。

#include <cusp/hyb_matrix.h>
#include <cusp/io/matrix_market.h>
#include <cusp/krylov/cg.h>

int main(void)
{
// create an empty sparse matrix structure (HYB format)
cusp::hyb_matrix<int, float, cusp::device_memory> A;

// load a matrix stored in MatrixMarket format
cusp::io::read_matrix_market_file(A, "5pt_10x10.mtx");

// allocate storage for solution (x) and right hand side (b)
cusp::array1d<float, cusp::device_memory> x(A.num_rows, 0);
cusp::array1d<float, cusp::device_memory> b(A.num_rows, 1);

// solve the linear system A * x = b with the Conjugate Gradient method
cusp::krylov::cg(A, x, b);

return 0;
}

这个问题让我们有可能讨论Mathematica 8的编译功能。也可以调用MMA的mathlink接口主题。我希望这里的人们发现这个问题值得思考，有趣，值得思考。

BR

Answer 1

如果你想使用LibraryLink（使用LibraryFunctionLoad来访问动态库函数作为Mathematica下限），实际上没有太多讨论空间，LibraryFunctions可以接收机器双精度或机器整数的Mathematica张量，你就完成了

Mathematica MTensor格式是一个密集的数组，就像你在C中自然使用一样，所以如果CUSP使用其他格式，你需要编写一些胶水代码来表示。

有关详细信息，请参阅LibraryLink tutorial。

您需要特别注意Interaction with Mathematica页面中的“MTensors的内存管理”部分，并选择“共享”模式以通过引用传递Mathematica张量。

Mathematica的库函数功能

1 个答案: