与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倍。