我一直在尝试通过定时使用以下代码对数组元素进行缩放和求和的例程来了解在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
我注意到求和函数循环中有一个奇怪的地方:
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