4(編譯展開)

今天繼續介紹第4種常見技巧,編譯展開,代碼見本文最後,重要的部分有註釋說明。

我們知道對於循環,編譯器會自動進行展開,但是如何展開,這完全不可控,如果我們自行碼代碼,代碼會顯得臃腫難看,大段重複代碼,因此本為介紹了一種常見的方法進行這種代碼展開,可以寫到任意層次,我這裡只寫到了DO16,見代碼加粗的部分。

實驗表明在手動展開後,加上O3的編譯優化,依然能夠比不手動展開要快,大家可以實驗以下的代碼。

可能會有讀者問,這能節省多少時間,會有多少好處,那麼請讀者朋友們做實驗來驗證吧,一切真知來自實踐。如果經常閱讀高質量開源代碼會常常看到這個技巧,希望讀者在深入理解後在工作中多多采用。

有興趣的朋友還可以用不同層次(4,16,32,64)的展開,看看展開多少是最優的,並解釋原因,那恭喜您,您的境界又上了一個新的臺階了。

注:

1)裡面會使用到一些此前介紹過的代碼,因此新讀者建議閱讀此前系列的內容,鏈接在本文最後。

2)本文的續篇參見:http://blog.csdn.net/pennyliang/archive/2010/10/30/5975678.aspx

在用-O3編譯後,用objdump -d test_m1_o3觀察代碼的情況[兩段rdtsc之間的代碼,為主要計算過程的代碼]

400730:       83 fd 02                cmp    $0x2,%ebp

  400733:       89 c6                   mov    %eax,%esi
  400735:       4c 8d 63 fc             lea    0xfffffffffffffffc(%rbx),%r12 //0xfffffffffffffffc為16進制的 -4,未來會有專門博客介紹這段代碼的具體含義,不在本文展開。
  400739:       7e 21                   jle    40075c <main+0xac>
  40073b:       8d 45 fd                lea    0xfffffffffffffffd(%rbp),%eax
  40073e:       4c 8d 63 fc             lea    0xfffffffffffffffc(%rbx),%r12
  400742:       31 d2                   xor    %edx,%edx
  400744:       48 8d 48 01             lea    0x1(%rax),%rcx
  400748:       8b 44 93 04             mov    0x4(%rbx,%rdx,4),%eax
  40074c:       03 04 93                add    (%rbx,%rdx,4),%eax          //循環並沒有被展開      
  40074f:       89 44 93 08             mov    %eax,0x8(%rbx,%rdx,4)
  400753:       48 83 c2 01             add    $0x1,%rdx                        //相當於i++
  400757:       48 39 ca                cmp    %rcx,%rdx
  40075a:       75 ec                   jne    400748 <main+0x98>
     用objdump -d test_m3_o3觀察代碼的情況[兩段rdtsc之間的代碼,為主要計算過程的代碼]
 400726:       89 c7                   mov    %eax,%edi
  400728:       8d 45 0f                lea    0xf(%rbp),%eax
  40072b:       85 ed                   test   %ebp,%ebp
  40072d:       89 ea                   mov    %ebp,%edx
  40072f:       4d 8d 6c 24 fc          lea    0xfffffffffffffffc(%r12),%r13
  400734:       be 02 00 00 00          mov    $0x2,%esi
  400739:       0f 48 d0                cmovs  %eax,%edx
  40073c:       c1 fa 04                sar    $0x4,%edx
  40073f:       83 fa 02                cmp    $0x2,%edx
  400742:       7e 79                   jle    4007bd <main+0x10d>
  400744:       4d 8d 6c 24 fc          lea    0xfffffffffffffffc(%r12),%r13
  400749:       be 02 00 00 00          mov    $0x2,%esi
  40074e:       66 90                   xchg   %ax,%ax
  400750:       8b 43 04                mov    0x4(%rbx),%eax    //eax是累加器,可以看到明顯的代碼展開
  400753:       03 03                   add    (%rbx),%eax
  400755:       83 c6 10                add    $0x10,%esi
  400758:       89 43 08                mov    %eax,0x8(%rbx)
  40075b:       03 43 04                add    0x4(%rbx),%eax
  40075e:       89 43 0c                mov    %eax,0xc(%rbx)
  400761:       03 43 08                add    0x8(%rbx),%eax
  400764:       89 43 10                mov    %eax,0x10(%rbx)
  400767:       03 43 0c                add    0xc(%rbx),%eax
  40076a:       89 43 14                mov    %eax,0x14(%rbx)
  40076d:       03 43 10                add    0x10(%rbx),%eax
  400770:       89 43 18                mov    %eax,0x18(%rbx)
  400773:       03 43 14                add    0x14(%rbx),%eax
  400776:       89 43 1c                mov    %eax,0x1c(%rbx)
  400779:       03 43 18                add    0x18(%rbx),%eax
  40077c:       89 43 20                mov    %eax,0x20(%rbx)
  40077f:       03 43 1c                add    0x1c(%rbx),%eax
  400782:       89 43 24                mov    %eax,0x24(%rbx)
  400785:       03 43 20                add    0x20(%rbx),%eax
  400788:       89 43 28                mov    %eax,0x28(%rbx)
  40078b:       03 43 24                add    0x24(%rbx),%eax
  40078e:       89 43 2c                mov    %eax,0x2c(%rbx)
  400791:       03 43 28                add    0x28(%rbx),%eax
  400794:       89 43 30                mov    %eax,0x30(%rbx)
  400797:       03 43 2c                add    0x2c(%rbx),%eax
  40079a:       89 43 34                mov    %eax,0x34(%rbx)
  40079d:       03 43 30                add    0x30(%rbx),%eax
  4007a0:       89 43 38                mov    %eax,0x38(%rbx)
  4007a3:       03 43 34                add    0x34(%rbx),%eax
  4007a6:       89 43 3c                mov    %eax,0x3c(%rbx)
  4007a9:       03 43 38                add    0x38(%rbx),%eax
  4007ac:       89 43 40                mov    %eax,0x40(%rbx)
  4007af:       03 43 3c                add    0x3c(%rbx),%eax
  4007b2:       89 43 44                mov    %eax,0x44(%rbx)
  4penny:       48 83 c3 40             add    $0x40,%rbx
  4007b9:       39 f2                   cmp    %esi,%edx
  4007bb:       7f 93                   jg     400750 <main+0xa0>
  4007bd:       39 f5                   cmp    %esi,%ebp
  4007bf:       7e 27                   jle    4007e8 <main+0x138>
  4007c1:       48 63 c6                movslq %esi,%rax
  4007c4:       48 c1 e0 02             shl    $0x2,%rax
  4liang:       49 8d 4c 05 00          lea    0x0(%r13,%rax,1),%rcx
  4007cd:       49 8d 54 04 f8          lea    0xfffffffffffffff8(%r12,%rax,1),%rdx
  4007d2:       8b 01                   mov    (%rcx),%eax
  4007d4:       03 02                   add    (%rdx),%eax
  4007d6:       83 c6 01                add    $0x1,%esi
  4007d9:       48 83 c1 04             add    $0x4,%rcx
  4007dd:       89 42 08                mov    %eax,0x8(%rdx)
  4007e0:       48 83 c2 04             add    $0x4,%rdx
  4007e4:       39 f5                   cmp    %esi,%ebp
  4007e6:       7f ea                   jg     4007d2 <main+0x122>

明顯看出展開後的代碼,另外從編譯出的可執行程序的大小也可看出這種差異,越是內聯展開的代碼可執行程序越大。

------------------------------------------編譯方法---------------------------------

在debug模式下的編譯


g++ -g  test.cpp -o test_m1 -D M_1

g++ -g  test.cpp -o test_m2 -D M_2

g++ -g  test.cpp -o test_m3 -D M_3

在-O3條件下的優化編譯:

g++ -O3  test.cpp -o test_m1_o3 -D M_1

g++ -O3  test.cpp -o test_m2_o3 -D M_2

g++ -O3  test.cpp -o test_m3_o3 -D M_3

-----------------------------------------運行方法-------------------------------------

./test_m1 1000000 //求1000000的斐波拉契數

./test_m2 1000000

./test_m3 1000000

-------------------------------------------代碼-------------------------------------

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define DO(x)  x
#define DO4(x)  x  x  x  x
#define DO8(x)  DO4(x) DO4(x)
#define DO16(x) DO8(x) DO8(x)
const int MAX = 512*1024*1024;
const float CPU_MHZ = 3000.164; //use cat /proc/cpuinfo get the value
const float CPU_tick_count_per_msecond = CPU_MHZ*1000;
#if defined(__i386__)

static __inline__ unsigned long long rdtsc(void)
{
  unsigned long long int x;
     __asm__ volatile ("rdtsc" : "=A" (x));
     return x;
}
#elif defined(__x86_64__)
static __inline__ unsigned long long rdtsc(void)
{
  unsigned hi, lo;
  __asm__ __volatile__ ("rdtsc" : "=a"(lo), "=d"(hi));
  return ( (unsigned long long)lo)|( ((unsigned long long)hi)<<32 );
}

#endif

int main(int argc,char **argv)
{
        if(argc!=2)
        {
                printf("command penny line: test N,N no more than%d/n",MAX);
                return 0;
        }
        
        int* F = (int*)malloc(MAX*sizeof(int));
        memset(F,0,MAX*sizeof(int)); //just warm up cache, to make calculate more accurate!!!   
        F[0]=1;
        F[1]=1;
        int Fx=atoi(argv[1]);
        int start = 0;
        int end = 0;
        start = rdtsc();
        #ifdef M_1
        for(int i=2;i<Fx;++i)        //通過循環的方法,遞進地計算
        {
                F[i]=F[i-1]+F[i-2];
        }
        #endif


        #ifdef M_2
        int r= Fx%4;
        int idx = Fx/4;
        int i =2;

        int j=0;
        for(;j<idx;++j)
        {
                DO4(F[i]=F[i-1]+F[i-2];i++;);  //通過循環展開的方法,展成4段代碼,循環規模降低到原來的1/4
        }
        for(;i<Fx;i++)
        {
                F[i]=F[i-1]+F[i-2];
        }
        #endif

        #ifdef M_3
        int r= Fx%16;   
        int idx = Fx/16;
        int i=2;

        int j=0;
        for(;j<idx;++j)
        {
                DO16(F[i]=F[i-1]+F[i-2];i++;); //展開成16段代碼
        }
        for(;i<Fx;i++)
        {
                F[i]=F[i-1]+F[i-2];
        }
        #endif
        end = rdtsc();
        printf("run tick count:%d/n",(end -start));
        printf("ret:%d/n",F[Fx-1]);
        free(F);
        return 0;
}

(續)

編譯展開的這篇博客被CSDN推了首頁http://blog.csdn.net/pennyliang/archive/2010/10/28/5971059.aspx,有些讀者反映有些太難,考慮到有些地方沒有講得太清楚,本文一併進行深入討論。

首先關鍵的代碼是:

#define DO(x)  x
#define DO4(x)  x  x  x  x
#define DO8(x)  DO4(x) DO4(x)
#define DO16(x) DO8(x) DO8(x)

讀者可以按葫蘆畫瓢,繼續展開,這並不難理解。

我們選取了計算斐波那契數作為例子,將代碼展開為16的倍數,但是因為Fx是用戶輸入的,可能是16的倍數也可能不是,因此需要做一些轉換,將Fx變為Fx=16idx+r,這樣可以確保可以做idx次展開,最後尾部再用一個循環計算完,如下:

int r= Fx%16;   
int idx = Fx/16;
int i=2;
for(;i<idx;)
{
    DO16(F[i]=F[i-1]+F[i-2];i++;); //展開成16段代碼
}
for(;i<Fx;i++)
{
    F[i]=F[i-1]+F[i-2];
}
如果我們不用編譯來做循環展開,我們的代碼可能就得是

for(;i<idx;)
{
    F[i]=F[i-1]+F[i-2];i++;

    F[i]=F[i-1]+F[i-2];i++;

    F[i]=F[i-1]+F[i-2];i++;

    ....寫16遍相同的代碼,多難看啊。
}
for(;i<Fx;i++)
{
    F[i]=F[i-1]+F[i-2];
}

第二,我們要特別注意memset的使用,memset在計時之前進行,是因為這樣的計算更為準確,malloc分配大內存是採用mmap的方式,即只分配虛地址,而沒有實際的調頁,而我們做一個memset是為了調頁,大家可以做這樣的實驗,做兩次相同的malloc,用rdtsc的方式進行計時,你會發現第一次malloc會慢一些,而第二次會快,因為第一次有調頁,而第二次幾乎無調頁(內存要足夠大,否則可能會swap,也可能有少量調頁)。

用memset相信也不難理解,因為這個時間混在裡面可能會看不出誤差,假定某市一季度GDP為10,二季度為15,看上去提高了50%,但是因為在計算過程中,有5份的GDP(可以想象成memset的代價)是多算在裡面的,如果扣除這5份,則實際上是從5提升了100%。因此我們在設計實驗時,需要把一些幹擾項扣除,來比較,實驗的數據才具有價值。

第三,我的實驗結果是展開為4次是比較快的,我認為有這樣一些主要原因,但還需要設計進一步的實驗來證明:

(1)編譯展開的層次過大,代碼會變大,而代碼存儲在文件中,進入執行需要有一個讀磁盤的過程,另外代碼大,代碼局部性就不強,L1 cache的一部分是存放代碼段的,如果代碼越小越緊湊,那麼執行起來就會越開,展開出1024層,代碼的局部性肯定很差。

(2)編譯展開的層次過小,代碼雖然緊湊,但是流水線不通暢,跳轉太多,因此也容易導致性能變低。

因此總會有一種展開的層次可以trade off的流水線和代碼段的緊湊性,我的實驗結果是展開到4層是最快的,這是受機器環境影響的,不知道其他同學的實驗結果是怎樣的。

最後提到的細節是DO16(F[i]=F[i-1]+F[i-2];i++;); 的最後一個分號是不必要的,但是添加在這裡是為了讓代碼更自然,相信大部分細心的讀者都能看到這一點。


书籍推荐