我一直在尝试通过使用以下代码对数组元素进行缩放和求和的例程定时来了解在 L1 缓存中拥有数组与内存的影响(我知道我应该只通过'来缩放结果a' 在最后;关键是在循环中同时进行乘法和加法 - 到目前为止,编译器还没有想出分解'a'):
double sum(double a,double* X,int size)
{
double total = 0.0;
for(int i = 0; i < size; ++i)
{
total += a*X[i];
}
return total;
}
#define KB 1024
int main()
{
//Approximately half the L1 cache size of my machine
int operand_size = (32*KB)/(sizeof(double)*2);
printf("Operand size: %d\n", operand_size);
double* X = new double[operand_size];
fill(X,operand_size);
double seconds = timer();
double result;
int n_iterations = 100000;
for(int i = 0; i < n_iterations; ++i)
{
result = sum(3.5,X,operand_size);
//result += rand();
}
seconds = timer() - seconds;
double mflops = 2e-6*double(n_iterations*operand_size)/seconds;
printf("Vector size %d: mflops=%.1f, result=%.1f\n",operand_size,mflops,result);
return 0;
}
请注意,为简洁起见,不包括 timer() 和 fill() 例程;如果您想运行代码,可以在此处找到它们的完整源代码:
现在,这就是有趣的地方。这是输出:
Operand size: 2048
Vector size 2048: mflops=588.8, result=-67.8
这完全是未缓存的性能,尽管 X 的所有元素都应该在循环迭代之间保存在缓存中。查看由以下生成的汇编代码:
g++ -O3 -S -fno-asynchronous-unwind-tables register_opt_example.cpp
我注意到 sum 函数循环中的一个奇怪之处:
L55:
movsd (%r12,%rax,8), %xmm0
mulsd %xmm1, %xmm0
addsd -72(%rbp), %xmm0
movsd %xmm0, -72(%rbp)
incq %rax
cmpq $2048, %rax
jne L55
说明:
addsd -72(%rbp), %xmm0
movsd %xmm0, -72(%rbp)
表示它将 sum() 中“total”的值存储在堆栈上,并在每次循环迭代时读取和写入它。我修改了程序集,使这个操作数保存在一个寄存器中:
...
addsd %xmm0, %xmm3
...
这个微小的变化创造了巨大的性能提升:
Operand size: 2048
Vector size 2048: mflops=1958.9, result=-67.8
tl; dr 我的问题是:为什么用寄存器替换单个内存位置访问会大大加快代码速度,因为单个位置应该存储在 L1 缓存中?哪些架构因素使这成为可能?重复写入一个堆栈位置会完全破坏缓存的有效性,这似乎很奇怪。
附录
我的 gcc 版本是:
Target: i686-apple-darwin10
Configured with: /var/tmp/gcc/gcc-5646.1~2/src/configure --disable-checking --enable-werror --prefix=/usr --mandir=/share/man --enable-languages=c,objc,c++,obj-c++ --program-transform-name=/^[cg][^.-]*$/s/$/-4.2/ --with-slibdir=/usr/lib --build=i686-apple-darwin10 --with-gxx-include-dir=/include/c++/4.2.1 --program-prefix=i686-apple-darwin10- --host=x86_64-apple-darwin10 --target=i686-apple-darwin10
Thread model: posix
gcc version 4.2.1 (Apple Inc. build 5646) (dot 1)
我的CPU是:
英特尔至强 X5650