如何使用valarray SIMD支持來優化此代碼？

Question

或使用SIMD指令？

#include <iostream> 
#include <string> 

#define sz 10000; 

typedef struct pixel{ 
    int b,g,r; 
}; 

int main() 
{ 
    pixel * src = new pixel[sz]; // arrays filled 
    pixel * ker = new pixel[sz]; // by random values 

    pixel px; 
    px.r=px.g=px.b=0; 
    for (int i=0; i<sz; i++){ 
     px.r+=src[i].r*ker[i].r; 
     px.g+=src[i].r*ker[i].g; 
     px.b+=src[i].r*ker[i].b; 
    } 


    return 0; 
} 

src - 包含一些隨機值的數組。

ker-like src，也包含隨機值。

的用於循環應該優化

Answer 1

你可以這樣做：

#include <valarray> 

#define sz 1000 

struct pixel { int r, g, b; }; 

int main() 
{ 
    std::valarray<int> src(3 * sz); 
    std::valarray<int> ker(3 * sz); 

    std::valarray<int> t = src * ker; 

    pixel px; 
    px.r = t[t % 3 == 0].sum(); 
    px.g = t[t % 3 == 1].sum(); 
    px.b = t[t % 3 == 2].sum(); 

    return 0; 
} 

我這裏假設你爲了rgb rgb rgb rgb在src和ker顏色值。

鏈接到文檔http://en.cppreference.com/w/cpp/numeric/valarray