7

最大値を見つけるために次のコードがあります

int length = 2000;
float *data;
// data is allocated and initialized

float max = 0.0;
for(int i = 0; i < length; i++)
{
   if(data[i] > max)
   {
      max = data;
   }
}

SSE3組み込み関数を使用してベクトル化を試みましたが、比較をどのように行うべきかについて、ある種の印象を受けました。

int length = 2000;
float *data;
// data is allocated and initialized

float max = 0.0;
// for time being just assume that length is always mod 4
for(int i = 0; i < length; i+=4)
{
  __m128 a = _mm_loadu_ps(data[i]);
  __m128 b = _mm_load1_ps(max);

  __m128 gt = _mm_cmpgt_ps(a,b);

  // Kinda of struck on what to do next
}

誰かがそれについていくつかのアイデアを与えることができますか？

4

1 に答える 1

11

したがって、コードはfloatの固定長配列で最大値を見つけます。わかった。

_mm_max_psがあり、それぞれ4つのフロートの2つのベクトルからペアワイズ最大値を取得します。では、これはどうですか？

int length = 2000;
float *data; // maybe you should just use the SSE type here to avoid copying later
// data is allocated and initialized

// for time being just assume that length is always mod 4
__m128 max = _mm_loadu_ps(data); // load the first 4
for(int i = 4; i < length; i+=4)
{
  __m128 cur = _mm_loadu_ps(data + i);
  max = _mm_max_ps(max, cur);
}

最後に、の4つの値のうち最大のものを取得しますmax（これについては、SSEを使用した__m128iベクトルの最大値の取得を参照してください）。

これは次のように機能するはずです。

ステップ1：

[43, 29, 58, 94] (this is max)
[82, 83, 10, 88]
[19, 39, 85, 77]

ステップ2：

[82, 83, 58, 94] (this is max)
[19, 39, 85, 77]

ステップ2：

[82, 83, 85, 94] (this is max)
于 2013-03-06T04:43:21.923 に答える