python - 我可以将 numpy 与 AMD 的 gpu 加速 blas 库链接起来吗

Question

我发现 numpy 可以与 blas 链接，我想到了为什么不使用 gpu 加速的 blas 库。

有人用过吗？

Answer 1

更新(2014-05-22)

AMD 发布了 AMD 核心数学库 (ACML) 6.0 版的测试版，可以通过在内部使用 clMath 将 FFT 和 BLAS 函数卸载到 GPU。公告在这里:ACML Beta 6.0 Release Leverages the Power of Heterogeneous Compute .这里需要注意的是，输入数据必须从 CPU 传输到 GPU，输出数据必须在每个 BLAS 或 FFT 调用时返回到 CPU。因此，当问题大到 ACML 将使用 GPU 而不是 CPU 时，AMD 有一堆用于调整的脚本。

为了完整起见，我还要提到 Nvidia 的 nvBLAS 库支持类似的功能，但它依赖于 cuBLAS 和 CUDA，因此它只能在 Nvidia GPU 上运行。

---

原始答案

不幸的是，AMD 的 GPU 加速 BLAS 库无法直接链接到 Numpy 或任何其他需要基于标准 CPU 的 BLAS 库的应用程序。原因是现有的GPU BLAS库都需要先将矩阵复制到GPU，然后再调用BLAS函数。这需要有人修改 Numpy 来执行此复制。

编辑:CLyther 看起来可以替代 Numpy 的某些功能并将所有内容转换为 OpenCL。看这里:http://srossross.github.io/Clyther/for_numpy_users.html