matlab - 使用 GPU 计算稀疏数据

Question

我正在 Matlab 中计算函数 f(x) = exp(-x)，其中 x 是标量向量。该函数是在 GPU 上计算的，例如

x_cpu = [4 5 11 1];
x = gpuArray(x_cpu);
f = exp(-x);

那么结果就是:

f = exp(-[4, 5, 11, 1]) = [0.183, 0.0067, 1.6702e-005, 0.3679]。

注意 f(x(3)) = f(11) = exp(-11) = 1.6702e-005 = 0.000016702，这是一个很小的值。因此，我想通过简单地设置 f(x(i)) = 来避免计算所有 x(i) > 10 的函数0.

我或许可以使用 x 的稀疏矩阵表示。但是，Parallel Computing Toolbox 不支持在 GPU 上对稀疏矩阵进行运算。

你会如何处理这个问题？

Answer 1

虽然并行计算工具箱不支持 GPU 上的稀疏矩阵运算，但 Jacket 支持。因此，一种可能的方法是简单地使用不同的工具。

免责声明是我在 Jacket 上工作，但我确实认为这对你有好处，因为它支持你想做的事情，而 PCT 不会做，并且对于 reasons here .