Java中的向量距离计算-优化

Question

作为图像处理功能的一部分，我需要计算两者之间的平方和 图像中有两行。

这部分代码占用了 96% 的运行时间：

for(int dx=0;dx<size;dx++) {
    int left = a[pa+dx];
    int right = b[pb+dx];
    int diff = (left & 0xFF) - (right & 0xFF);
    sum += diff*diff;
}

地点：

• a、b 属于 byte[] 类型
• sum 是 long
• size 是 int 并且通常具有较大的值（大约 400）

运行 Java 7 64 位。我试图用 a[pa++] 之类的东西替换 a[pa+dx] 性能也好不到哪里去。

执行保存运行的代码与用 C++ 编写的代码完全相同 整体快两倍（！）和 据我所知，应该没有重要的理由为什么不使用这个 Java 代码要一样快，尤其是当边界检查可以通过以下方式移出循环时 编译器。

如何优化这些东西以使其与 C++ 代码一样好 - 最终它是 整数运算在 Java 中应该不会慢很多

编辑：C++ 示例如下：

unsigned char const *srcptr=&a[pa];
unsigned char const *tgtptr=&b[pb];
for(int dx=0;dx < size;dx++) {
    int p1=*srcptr++;
    int p2=*tgtptr++;
    int diff = p1 - p2;
    sum += diff * diff;
}

我想了解如何使用 HotSpot 优化器来创建代码 和上面显示的 C++ 代码一样快，最后还是相当简单易行 优化线条。

Answer 1

它只是很小，但您不需要& 0xFF 来计算差异：差异将是相同的有符号或无符号。

100 - -1 = 101  // signed
228 - 127 = 101 // unsigned

那就是更紧的循环体：

for (int dx = 0; dx < size; dx++) {
    int diff = a[pa+dx] - b[pb+dx];
    sum += diff*diff;
}

编辑：

关于有符号和无符号字节算术似乎有些混淆。如果您怀疑它们是否相同，请执行以下操作：

byte a = -128;
byte b = 127;
int diff = a - b;
System.out.println(diff); // -255

a = 127;
b = -128;
diff = a - b;
System.out.println(diff); // 255

原因diff值的范围大于byte(-128..127)，是java自动将byte加宽到int之前 计算因为目标变量是int。

Answer 2

将& 0xFF 移到循环之外。

通过计算a 和b 的int[]-version 并使用它们重写您的循环来做到这一点。

Answer 3

在我使用不同的 C++ 编译器和不同的 Java 版本测试了相同的算法后，我得出的结论是 GCC 是非常优秀的编译器，它比 intel 和 clang 更好地优化了代码！

这些是用 C++ 和 Java 实现的相同算法的运行时间（当上面的行是运行时间的 96% 时：

Intel 12.1  1:58
GCC 4.6     0:43
GCC 4.4     0:43
Clang       1:20
Java 7      1:20
Java 6      1:23

这表明 Java 的运行速度与 clang 一样快，而英特尔编译器由于某种原因做得非常糟糕，但是 gcc 给出了最好的结果，所以我真的不能指望 Java 的运行速度比 大多数 C++ 编译器都可以。

注意这是 gcc 生成的程序集：

.L225:
    movzbl  (%rcx), %r8d
    movzbl  (%rsi), %r10d
    addl    $1, %edx
    addq    $1, %rcx
    addq    $1, %rsi
    subl    %r10d, %r8d
    imull   %r8d, %r8d
    movslq  %r8d, %r8
    addq    %r8, %rax
    cmpl    %edx, %ebp
    ja      .L225

还有这个是由clang生成的：

.LBB0_26:
    movzbl  (%r11), %r13d
    movzbl  (%r14), %esi
    subl    %r13d, %esi
    imull   %esi, %esi
    movslq  %esi, %rsi
    addq    %rsi, %rcx
    incq    %r11
    incq    %r14
    decq    %r12
    jne     .LBB0_26

有什么区别？ GCC 重新排列指令，以便它们可以 在管道中并行运行，例如：

    movzbl  (%rcx), %r8d
    movzbl  (%rsi), %r10d
    addl    $1, %edx
    addq    $1, %rcx
    addq    $1, %rsi

底线，Java 运行时间很好。

编辑：在向 Intel 编译器提供 -xHost 选项（针对当前 CPU 进行优化）后，运行时间提高到 56 秒（使用 mmx 指令），但仍然不如 gcc，但好不了多少比Java

Answer 4

如果数组a或b的大小为“size”，则可以避免for条件：

try{
    for (int dx = 0; ; dx++) {
        ...
        ...
    }
}catch(ArrayIndexOutOfBoundException e){}

两条线是直线还是曲线？您可以发布问题的图形表示，还是数组的数字示例？也许有更好的几何解决方案？