15

你怎么用__m256d？

假设我想在具有 3-64 位精度组件（、、和）_mm256_add_pd的简单类上使用英特尔 AVX 指令。使用它的正确方法是什么？Vector3doublexyz

由于x和是该类的成员y， _我可以用变量声明它们吗？zVector3union__m256d

union Vector3
{
  struct { double x,y,z ; } ;
  __m256d _register ;  // the Intel register?
} ;

那我可以去吗：

Vector3 add( const Vector3& o )
{
  Vector3 result;
  result._register = _mm256_add_pd( _register, o._register ) ; // add 'em
  return result; 
}

这行得通吗？还是我需要申报临时工，

Vector3 add( const Vector3& o )
{
  __m256d d1 = *(__m256d*)(&x) ; // ? Cast to __m256d?
  __m256d d2 = *(__m256d*)(&o.x) ; // ? Cast to __m256d?
  __m256d result = _mm256_add_pd( d1, d2 ) ; // add 'em
  return Vector3( result ) ; // make a ctor that accepts __m256d?
}

编辑

我想出了这个例子，

#include <stdio.h>
#include <intrin.h>

int main()
{
  __m256d a, b, res;

  for( int i = 0; i < sizeof(__m256d)/sizeof(double); i++ )
  {
    a.m256d_f64[i] = i ;
    b.m256d_f64[i] = 2*i ;
  }

  // Perform __4__ adds.
  res = _mm256_add_pd(a, b);

  for( int i = 0; i < sizeof(__m256d)/sizeof(double); i++ )
  {
    printf("%f + %f = %f\n", a.m256d_f64[i], b.m256d_f64[i], res.m256d_f64[i]);
  }
  puts("");
}

我想现在的问题是，是否会 自动_mm256_add_pd执行加载操作，或者如果我不将我的寄存器声明为接近使用它们的地方的本地人，会不会出现问题？（我怕酒店房间/书桌抽屉类型问题）__m256d

编辑2：

我尝试__m256在我相当大的项目中添加一个寄存器，我得到了一大堆

错误 C2719：“值”：带有 __declspec(align('32')) 的形式参数将不会对齐

错误，它让我相信你不能在一个类中保留__m256寄存器，而是应该将它们声明为本地人？

4

1 回答 1

16

首先，我想澄清一点困惑。 __m256d不是一种寄存器，它是一种可以加载到 AVX 寄存器中的数据类型。A__m256d不再是一个寄存器，正如一个int是一个寄存器。有几种方法可以将数据输入和输出__m256d（或任何其他向量类型）：

使用 a union：是的，union诀窍有效。它工作得很好，因为联合通常会有正确的对齐方式（尽管malloc可能不会，使用posix_memalignor _aligned_malloc）。

class Vector3 {
public:
    Vector3(double xx, double yy, double zz);
    Vector3(__m256d vvec);


    Vector3 operator+(const Vector3 &other) const
    {
        return Vector3(_mm256_add_pd(vec, other.vec));
    }

    union {
        struct {
            double x, y, z;
        };
        __m256d vec; // a data field, maybe a register, maybe not
    };
};

使用内在函数：在函数内部，使用内在函数将数据传入和传出向量类型通常更容易。

__m256d vec = ...;
double x, y, z;
vec = _mm256_add_pd(vec, _mm256_set_pd(x, y, z, 0.0));

使用指针强制转换：出于几个原因，强制转换指针是最后的手段。

  1. 指针可能未正确对齐。

  2. 强制转换指针有时会干扰编译器的别名分析。

  3. 指针转换绕过了许多安全保证。

所以我只会使用指针转换来浏览大量数据。

于 2012-10-14T01:11:58.627 回答