使用rdtsc指令，测量程序的运行速度 (运行CPU周期数）

 

 

linux开发版最长问到的问题之一就是在linux下如何得到更精确的计时。其实有很多办法，比如以前就有人用select。不过现在有更精确的实时时钟可以用，这就是用CLOCK_PROCESS_CPUTIME_ID做参数通过timer_create来创建timer。通过clock_getres可以得到系统的精度。

实际上，上述方法底层用的就是CPU的rdtsc指令。 在linux下，可以用到CPU的rdtsc指令（当然，windows下也是这样。现在的CPU一般都支持这个指令）来得到一个CPU的时间戳（Time   Stamp），这个值是每个指令周期都增加的，再根据CPU的频率就可以计算出时间来。用这种方式，对于频率高的CPU，甚至可以实现纳秒级的时间控制。不过据说因为精度太高，这种方式数据抖动比较厉害，每次结果都不一样，经常有几百甚至上千的差距。不过在我的实验中误差并没有这么大，上下在1%左右，不过如果你的要求高的话，这1%也是很厉害的。不过sleep调用也是会有误差的，这里就不去研究了。

下面是源码：

 

#include  < stdio.h >

int  get_rdtsc()  ... {
  asm("rdtsc");
}

int  main()  ... {
  int i;
  for(i=0;i<10;i++) ...{
    int t1 = get_rdtsc();
    sleep(1);
    int t2 = get_rdtsc();
    printf("t2 - t1 = %ld (%ldMHZ) ", t2 - t1, (t2-t1)/1000000);
  }
}

 

 

我的实验环境：Intel E6320/2G，使用VMWare虚拟机，系统是centos5-x86。下面是输出结果。

t2 - t1 = 1865320163 (1865MHZ)
t2 - t1 = 1865996702 (1865MHZ)
t2 - t1 = 1862758710 (1862MHZ)
t2 - t1 = 1865247214 (1865MHZ)
t2 - t1 = 1863456686 (1863MHZ)
t2 - t1 = 1865427930 (1865MHZ)
t2 - t1 = 1865423429 (1865MHZ)
t2 - t1 = 1863616999 (1863MHZ)
t2 - t1 = 1865105795 (1865MHZ)
t2 - t1 = 1867224090 (1867MHZ) 

 

2011-09-10 16:21:11

分类： C/C++

    最近发现了rdtsc指令，可以获取CPU指令周期数，喜出望外，wiki了下相关的知识，写了代码利用CPU周期来测量程序的运行时间。

 

    rdtsc指令返回的是自开机始CPU的周期数，返回的是一个64位的值EDX：EAX（高32在EDX，低32位在EAX）。OK，完全可以利用这条指令，测试我们的关注的一段代码的执行效率。

    题外话，我兴冲冲的告诉我老大，我发现了一个测量程序性能的好办法，老大淡然的说，不会是rdtsc吧。呵呵我和老大的水平差距还是云泥之别啊。他告诉我可以去Linux Kernel查看内核是如何做的。呵呵，我就照搬了kernel的实现。兼练习以下C和汇编混合编程。

    在网上搜索相关的资源看到了陈硕大牛的多核时代不宜再用 x86 的 RDTSC 指令测试指令周期和时间 他给出了几个理由，有兴趣的兄弟可以去读下。

    1多核，不能保证每个核的TSC是一样的。

    2 CPU的时钟频率可变

    3 乱序执行导致测量不准。

    

    文献5指出，指令可能乱序执行，并给出个例子，下面的一段代码，本意是测量fdiv需要的CPU周期，但是，由于乱序执行，第二个rdtsc指令可能在fdiv之前执行，造成，无法测量fdiv的需要的CPU周期。文献给出了解决办法。有兴趣的兄弟可以去阅读参考文献5.

        rdtsc                   ; read time stamp

        mov     time, eax       ; move counter into variable

        fdiv                    ; floating-point divide

        rdtsc                   ; read time stamp

        sub     eax, time       ; find the differenc

参考文献：

 1   深入理解计算机系统

 2   Linux Kernel code

 3   wiki

 4   多核时代不宜再用 x86 的 RDTSC 指令测试指令周期和时间

 5    http://www.ccsl.carleton.ca/~jamuir/rdtscpm1.pdf

    

2. #include<stdio.h> 
4. #include<stdlib.h>
5.  #include<linux/types.h>
6. 
   
7. #define TIMES 100
   
8. #define SIZE 1024
   
9. 
   
10. __u64 rdtsc()
    
11. {
    
12.         __u32 lo,hi;
    
13. 
    
14.         __asm__ __volatile__
    
15.         (
    
16.          "rdtsc":"=a"(lo),"=d"(hi)
    
17.         );
    
18.         return (__u64)hi<<32|lo;
    
19. }
    
20. 
    
21. int myfunction()
    
22. {
    
23.         int i;
    
24.         char *p = NULL;
    
25.         for(i = 0;i<TIMES;i++)
    
26.         {
    
27.                 p = (char*)malloc(SIZE*sizeof(char));
    
28.                 if(p)
    
29.                 {
30.                     free(p);
    
31.                 }
32.                 else
    
33.                 {
    
34.                     printf("malloc failed when i = %d\n",i);
    
35.                 }
    
36.         }
    
37.         return 0;
    
38. }
    
39. int test_rdtsc()
    
40. {
    
41.         __u64 begin;
    
42.         __u64 end;
    
43. 
    
44.         begin = rdtsc();
    
45.         myfunction();
    
46.         end = rdtsc();
    
47.         printf("myfunction cost %llu CPU cycles\n",end-begin);
    
48.         return 0;
    
49. }
    
50. 
    
51. int main()
    
52. {
    
53.        test_rdtsc();
    
54.         return 0;
    
55. }

    执行结果如下

    

1. root@libin:~/program/assembly/rdtsc# ./test
   
2. myfunction cost 310949 CPU cycles