基于Intel处理器的PREFETCH指令的基本使用

以下这个例子很简单,分别有两个双精度浮点类型的二维数组:double[512][1024],共4MB数据,分为512行1024列,每列位8个字节。现在将每对浮点数组的每一行的第一个元素相加,然后将结果存到一个在栈上分配的数组。

算法的汇编文件如下所示:

.align 2
.text
.globl _fast_int_sqrt
.globl _get_cycles
.globl _naive_calc
.globl _opt_calc

_get_cycles:

rdtsc
shl $32, %rdx
or %rdx, %rax
ret

_naive_calc:

mov $100, %r9

native_loop:

mov $512, %r8
xor %rcx, %rcx
xor %rax, %rax

naive_process:

fldl (%rsi, %rcx)
fldl (%rdx, %rcx)

fadd %st(0), %st(1)
fcmove %st(1), %st(0)
faddp %st(0), %st(1)


fstpl (%rdi, %rax)

add $(8 * 1024), %rcx
add $8, %rax
sub $1, %r8
jne naive_process

sub $1, %r9
jne native_loop

ret

_opt_calc:

mov $100, %r9

opt_loop:

mov $512, %r8
xor %rcx, %rcx
xor %rax, %rax

opt_process:

prefetcht0 (8 * 1024)(%rsi, %rcx)
prefetcht0 (8 * 1024)(%rdx, %rcx)

fldl (%rsi, %rcx)
fldl (%rdx, %rcx)

fadd %st(0), %st(1)
fcmove %st(1), %st(0)
faddp %st(0), %st(1)


fstpl (%rdi, %rax)

add $(8 * 1024), %rcx
add $8, %rax
sub $1, %r8
jne opt_process

sub $1, %r9
jne opt_loop

ret

上面可以看到opt_calc与naive_calc的唯一区别就是opt_calc多了两条PREFETCH指令。

下面是C语言的测试函数:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

extern unsigned long get_cycles(void);

// Calculate 2 * (src1 + src2)
extern void naive_calc(void *dst, const void *src1, const void *src2);
extern void opt_calc(void *dst, const void *src1, const void *src2);

extern void hey(int y);

void hey(int y)
{
    // Initialize the buffers
    void *src1 = malloc(8 * 1024 * 512);
    void *src2 = malloc(8 * 1024 * 512);
    double dst[512];
    
    memset(src1, 0, 8 * 1024 * 512);
    memset(src2, 0, 8 * 1024 * 512);
    
    double *p = (double*)src1;
    *p = 3.141;
    p = (double*)src2;
    *p = 2.163;
    
    // naive
    unsigned long t1 = get_cycles();
    
    naive_calc(dst, src1, src2);        // rdi, rsi, rdx
    
    unsigned long t2 = get_cycles();
    
    // opt
    unsigned long t3 = get_cycles();
    
    opt_calc(dst, src1, src2);
    
    unsigned long t4 = get_cycles();
    
    printf("Naive cycles is: %lu\n", t2 - t1);
    printf("Opt cycles: %lu\n", t4 - t3);
    printf("The value is: %f\n", dst[0]);
    
    free(src1);
    free(src2);
}

这里为了增加些计算周期,所以用了乘以2的方式。

俺在Mac Mini CPU为P7350,2.00GHz的双核基于45nm技术的Intel Core架构的处理器下测试,添加opt_calc的性能大约能增长8%到20%。 

posted @ 2012-03-09 03:39  zenny_chen  Views(5493)  Comments(0Edit  收藏  举报