天天看點

初識和c/c++調用SSE2

需要程式員程式設計的時候指明使用sse的方式,你可以看到很多媒體播放或處理軟體都有類似的選項讓你選擇。程式設計的方法類似于此(這裡是SSE的例子,4.1類似)

SSE是英特爾提出的即MMX之後新一代(當然是幾年前了)CPU指令集,最早應用在PIII系列CPU上。現在已經得到了Intel PIII、P4、Celeon、Xeon、AMD Athlon、duron等系列CPU的支援。而更新的SSE2指令集僅得到了P4系列CPU的支援,這也是為什麼這篇文章是講SSE而不是SSE2的原因之一。另一個原因就是SSE和SSE2的指令系統是非常相似的,SSE2比SSE多的僅是少量的額外浮點處理功能、64位浮點數運算支援和64位整數運算支援。

SSE為什麼會比傳統的浮點運算更快呢?因為它使用了128位的存儲單元,這對于32位的浮點數來講,是可以存下4個的,也就是說,SSE中的所有計算都是一次性針對4個浮點數來完成的,這種批處理當然就會帶來效率的提升。我們再來回顧一下SSE的全稱:Stream SIMD Extentions(流SIMD擴充)。SIMD就是single instruction multiple data,連起來就是“資料流單指令多資料擴充”,從名字我們就可以更好的了解SSE是如何工作的了。

雖然SSE從理論上來講要比傳統的浮點運算會快,但是他所受的限制也很多,首先,雖然他執行一次相當于四次,會比傳統的浮點運算執行4次的速度要快,但是他執行一次的速度卻并沒有想象中的那麼快,是以要展現SSE的速度,必須有Stream做前提,就是大量的流資料,這樣才能發揮SIMD的強大作用。其次,SSE支援的資料類型是4個32位(共計128位)浮點數集合,就是C、C++語言中的float[4],并且必須是以16位位元組邊界對齊的(稍後會以代碼來進行闡釋,關于邊界對齊的概念,讀者可以參考論壇上的其它文章,都會有很詳細的解答,我這裡就恕不贅述了)。是以這也給輸入和輸出帶來了不少的麻煩,實際上主要影響SSE發揮性能的就是不停的對資料進行複制以适用應它的資料格式。

我是一個C++程式員,對彙編并不很熟,但我又想用SSE來優化我的程式,我該怎麼做呢?幸好VC++.net為我們提供了很友善的指令C函數級的封裝和C格式資料類型,我們隻需像平時寫C++代碼一樣定義變量、調用函數就可以很好的應用SSE指令了。

當然了,我們需要包含一個頭檔案,這裡面包括了我們需要的資料類型和函數的聲明:

#include <xmmintrin.h>

SSE運算的标準資料類型隻有一個,就是:

__m128,它是這樣定義的:

typedef struct __declspec(intrin_type) __declspec(align(16)) __m128 {

float m128_f32[4];

} __m128;

簡化一下,就是:

struct __m128

{

float m128_f32[4];

};

比如要定義一個__m128變量,并為它賦四個float整數,可以這樣寫:

__m128 S1 = { 1.0f, 2.0f, 3,0f, 4,0f };

要改變其中第2個(基數為0)元素時可以這樣寫:

S1.m128_f32[2] = 6.0f;

令外我們還會用到幾個指派的指令,它可以讓我們更友善的使用這個資料結構:

S1 = _mm_set_ps1( 2.0f );

它會讓S1.m128_f32中的四個元素全部賦予2.0f,這樣會比你一個一個指派要快的多。

S1 = _mm_setzero_ps();

這會讓S1中的所有4個浮點數都置零。

還有一些其它的指派指令,但執行起來還沒有自己逐個指派來的快,隻做為一些特殊用途,如果你想了解更多的資訊,可以參考MSDN -> VisualC++參考 -> C/C++Language -> C++Language Reference -> Compiler Intrinsics -> MMX, SSE, and SSE2 Intrinsics -> Stream SIMD Extensions(SSE)章節。

一般來講,所有SSE指令函數都有3個部分組成,中間用下劃線隔開:

_mm_set_ps1

mm表示多媒體擴充指令集

set表示此函數的含義縮寫

ps1表示該函數對結果變量的影響,由兩個字母組成,第一個字母表示對結果變量的影響方式,p表示把結果做為指向一組資料的指針,每一個元素都将參與運算,S表示隻将結果變量中的第一個元素參與運算;第二個字母表示參與運算的資料類型。s表示32位浮點數,d表示64位浮點數,i32表示32位定點數,i64表示64位定點數,由于SSE隻支援32位浮點數的運算,是以你可能會在這些指令封裝函數中找不到包含非s修飾符的,但你可以在MMX和SSE2的指令集中去認識它們。

接下來我舉一個例子來說明SSE的指令函數是如何使用的,必須要說明的是我以下的代碼都是在VC7.1的平台上寫的,不保證對其它如Dev-C++、Borland C++等開發平台的完全相容。

為了友善對比速度,我會用常歸方法和SSE優化兩種寫法寫出,并會用一個測試速度的類CTimer來進行計時。

這個算法是對一組float值進行放大,函數ScaleValue1是使用SSE指令優化的,函數ScaleValue2則沒有。我們用10000個元素的float數組資料來測試這兩個算法,每個算法運算10000遍,下面是測試程式和結果:

#include <xmmintrin.h>

#include <windows.h>

class CTimer

{

public:

__forceinline CTimer( void )

   {

          QueryPerformanceFrequency( &m_Frequency );

          QueryPerformanceCounter( &m_StartCount );

   }

   __forceinline void Reset( void )

   {

          QueryPerformanceCounter( &m_StartCount );

   }

   __forceinline double End( void )

   {

          static __int64 nCurCount;

          QueryPerformanceCounter( (PLARGE_INTEGER)&nCurCount );

          return double( nCurCount * ( *(__int64*)&m_StartCount ) ) / double( *(__int64*)&m_Frequency );

   }
           

private:

LARGE_INTEGER m_Frequency;

   LARGE_INTEGER m_StartCount;
           

};

void ScaleValue1( float *pArray, DWORD dwCount, float fScale )

{

DWORD dwGroupCount = dwCount / 4;

   __m128 e_Scale = _mm_set_ps1( fScale );

   for ( DWORD i = 0; i < dwGroupCount; i++ )

   {

          *(__m128*)( pArray + i * 4 ) = _mm_mul_ps( *(__m128*)( pArray + i * 4 ), e_Scale );

   }
           

}

void ScaleValue2( float *pArray, DWORD dwCount, float fScale )

{

for ( DWORD i = 0; i < dwCount; i++ )

   {

          pArray[i] *= fScale;

   }
           

}

#define ARRAYCOUNT 10000

int __cdecl main()

{

float __declspec(align(16)) Array[ARRAYCOUNT];

   memset( Array, 0, sizeof(float) * ARRAYCOUNT );

   CTimer t;

   double dTime;

   t.Reset();



  for ( int i = 0; i < 100000; i++ )

   {

          ScaleValue1( Array, ARRAYCOUNT, 1000.0f );

   }

   dTime = t.End();

   cout << "Use SSE:" << dTime << "秒" << endl;

   t.Reset();

   for ( int i = 0; i < 100000; i++ )

   {

          ScaleValue2( Array, ARRAYCOUNT, 1000.0f );

   }

   dTime = t.End();

   cout << "Not Use SSE:" << dTime << "秒" << endl;

   system( "pause" );

   return 0;
           

}

Use SSE:0.997817

Not Use SSE:2.84963

這裡要注意一下,我使用了__declspec(align(16))做為數組定義的修釋符,這表示該數組是以16位元組為邊界對齊的,因為SSE指令隻能支援這種格式的記憶體資料。

我們在這裡看到了SSE算法的強大,相信它會成為多媒體程式員手中用來對付無窮盡流媒體資料的一把利劍。