fmaf*的用法相比,使用+函数使我的性能下降了很多。我在两台Linux机器上,并使用g++ 4.4.3和g++ 4.6.3

如果在不使用myOut的情况下填充了fmaf vector ,则在两台不同的计算机上,以下代码可以更快地运行。

g++ 4.6.3和Intel(R)Xeon(R)CPU E5-2650 @ 2.00GHz的服务器

$ ./a.out fmaf
Time: 1.55008 seconds.
$ ./a.out muladd
Time: 0.403018 seconds.

g++ 4.4.3和Intel(R)Xeon(R)CPU X5650 @ 2.67GHz的服务器
$ ./a.out fmaf
Time: 0.547544 seconds.
$ ./a.out muladd
Time: 0.34955 seconds.
fmaf版本(为了避免多余的舍入然后更加精确)是否应该更快?
#include <stddef.h>
#include <iostream>
#include <math.h>
#include <string.h>
#include <stdlib.h>

#include <sys/time.h>

int main(int argc, char** argv) {
  if (argc != 2) {
    std::cout << "missing parameter: 'muladd' or 'fmaf'"
              << std::endl;
    exit(-1);
  }
  struct timeval start,stop,result;
  const size_t mySize = 1e6*100;

  float* myA = new float[mySize];
  float* myB = new float[mySize];
  float* myC = new float[mySize];
  float* myOut = new float[mySize];

  gettimeofday(&start,NULL);
  if (!strcmp(argv[1], "muladd")) {
    for (size_t i = 0; i < mySize; ++i) {
      myOut[i] = myA[i]*myB[i]+myC[i];
    }
  } else if (!strcmp(argv[1], "fmaf")) {
    for (size_t i = 0; i < mySize; ++i) {
      myOut[i] = fmaf(myA[i], myB[i], myC[i]);
    }
  } else {
    std::cout << "specify 'muladd' or 'fmaf'" << std::endl;
    exit(-1);
  }

  gettimeofday(&stop,NULL);
  timersub(&stop,&start,&result);
  std::cout << "Time: " <<  result.tv_sec + result.tv_usec/1000.0/1000.0
            << " seconds." << std::endl;

  delete []myA;
  delete []myB;
  delete []myC;
  delete []myOut;
}

最佳答案

您的问题的答案称为向量化。使用g++ -O3 -S编译时,比较g++ 4.4.6生成的汇编代码,以了解代码的两个部分:
muladd部分:

.L10:
    movaps  %xmm2, %xmm0
    movaps  %xmm2, %xmm1
    movlps  (%rbx,%rax), %xmm0
    movlps  (%r12,%rax), %xmm1
    movhps  8(%rbx,%rax), %xmm0
    movhps  8(%r12,%rax), %xmm1
    mulps   %xmm1, %xmm0
    movaps  %xmm2, %xmm1
    movlps  0(%rbp,%rax), %xmm1
    movhps  8(%rbp,%rax), %xmm1
    addps   %xmm1, %xmm0
    movaps  %xmm0, 0(%r13,%rax)
    addq    $16, %rax
    cmpq    $400000000, %rax
    jne     .L10

所有这些*ps都对压缩的单精度数字执行操作。这些是SSE指令,因此每个包由每个数组的4个连续元素组成。

实现fmaf版本的循环是:
.L14:
    movss   (%rbx,%r14,4), %xmm0
    movss   0(%rbp,%r14,4), %xmm2
    movss   (%r12,%r14,4), %xmm1
    call    fmaf
    movss   %xmm0, 0(%r13,%r14,4)
    addq    $1, %r14
    cmpq    $100000000, %r14
    jne     .L14

在这里,标量SSE指令用于在一次将数据移动到一个数组元素上,每次迭代都会对fmaf进行函数调用。

循环的 vector 部分较长,但执行的迭代次数少4倍。

08-25 18:23