JBLAS的表现

Question

我想进行重矩阵乘法

由于我的应用程序需要高性能，我决定使用JBLAS。

但是，我发现JBLAS比测试中的简单for循环慢。

    double[][] M = new double[3000][3100];
    double[] a = new double [3100];
    double[] b = new double[3000];
    for(double[] row: M){
        Arrays.fill(row, 3.343);
    }
    Arrays.fill(a, 1.324);
    DoubleMatrix M1 = new DoubleMatrix(M);
    DoubleMatrix a1 = new DoubleMatrix(a);
    DoubleMatrix b1= new DoubleMatrix(b);

    //1. Simple for loop : 366 ms
    for(int i=0; i<3000; i++){
        for(int j=0; j<3100; j++){
            b[i] = b[i] + a[j]*M[i][j];
        }
    }
    // 2. JBLAS : 1190 ms
    b1 = M1.mmul(a1);

虽然它们的计算方法相同，但JBLAS比简单的循环慢3倍。

是由于我的错误吗？或其他原因？

谢谢！

Answer 1

我自己找到了解决方案，然后分享！

原因是本机代码涉及复制数据，因此矩阵向量乘法等操作无法从本机代码中受益。

因此，如果mmul函数识别矩阵*向量乘法，它使用Java代码。

您可以在此处获取更多信息 https://groups.google.com/forum/#!topic/jblas-users/HY2acEE3Y10