在Java中使用GPU执行计算

Question

我正在寻找一种能够成为Java程序一部分的方式，特别是需要进行许多重复的，稍微复杂的计算，在计算机的GPU上执行而不是单独执行在CPU上。我知道像LWJGL这样的库允许Java程序使用GPU，但我还没有找到一种简单的方法来简单地对它进行计算，而这些计算不一定与图形有关。

有没有办法，例如使用计算机的GPU加速以下代码？

    long t = System.currentTimeMillis();
    double x;
    for(int i=0; i < 1080; i++){
        for(int j=0; j<1920; j++){
            x = Math.sqrt(Math.tan(1d - Math.sin(i * j)));
        }
    }
    System.out.println(System.currentTimeMillis() - t);

当我测试上面的单CPU操作时，它需要大约1.25秒才能完成。

Answer 1

您可能想尝试aparapi。除此之外，任何围绕本机OpenCL / CUDA的Java包装器都可能有用。