您如何使用__m256d

假设我想在具有3-64位_mm256_add_pd精度组件(Vector3doublex)的简单y类上使用Intel AVX指令 z 。正确的使用方法是什么?

由于xyzVector3类的成员,因此_我可以使用union变量在__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寄存器声明为接近使用它们的本地变量,是否会弄乱某些东西? (我担心hotel room / deskdrawer类型问题)

编辑2:

我尝试将__m256寄存器添加到我的相当大的项目中,然后得到了一堆



错误,使我相信您不能将__m256寄存器保留在类内,而是应该将它们声明为locals?

最佳答案

首先,我想消除一些困惑。 __m256d不是寄存器的类型,它是可以加载到AVX寄存器中的数据类型。 __m256d不再是寄存器,而int只是寄存器。有几种方法可以将数据输入__m256d(或任何其他 vector 类型):

使用union:是的,union技巧有效。因为联合通常具有正确的对齐方式,所以它工作得很好(尽管malloc可能不正确,请使用posix_memalign_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
    };
};

使用内在函数:在函数内部,通常更容易使用内在函数来获取 vector 类型中的数据。
__m256d vec = ...;
double x, y, z;
vec = _mm256_add_pd(vec, _mm256_set_pd(x, y, z, 0.0));

使用指针强制转换:出于几个原因,强制转换指针是最后的手段。
  • 指针可能未正确对齐。
  • 强制转换指针有时可能会干扰编译器的别名分析。
  • 指针转换绕过了许多安全保证。

  • 因此,我只会使用指针转换来浏览大量数据。

    07-24 09:45
    查看更多