3

我有一些执行饱和算术运算的 for 循环。例如：

在我的情况下，饱和添加的实现如下：

    static void addsat(Vector &R, Vector &A, Vector &B)
    {
        int32_t a, b, r;
        int32_t max_add;
        int32_t min_add;


        const int32_t SAT_VALUE = (1<<(16-1))-1;
        const int32_t SAT_VALUE2 = (-SAT_VALUE - 1);
        const int32_t sat_cond = (SAT_VALUE <=  0x7fffffff);
        const uint32_t SAT = 0xffffffff >> 16;

        for (int i=0; i<R.length; i++)
        { 

            a = static_cast<uint32_t>(A.data[i]);
            b = static_cast<uint32_t>(B.data[i]);

            max_add = (int32_t)0x7fffffff - a;
            min_add = (int32_t)0x80000000 - a;

            r = (a>0 && b>max_add) ?  0x7fffffff : a + b;
            r = (a<0 && b<min_add) ?  0x80000000 : a + b;


            if ( sat_cond == 1)
            {
                std_max(r,r,SAT_VALUE2);
                std_min(r,r,SAT_VALUE);
            }
            else
            {
                r = static_cast<uint16_t> (static_cast<int32_t> (r));

            }


 R.data[i] = static_cast<uint16_t>(r);
    }
    }

我看到 x86 中存在 paddsat 固有特性，它可能是这个循环的完美解决方案。我确实将代码自动矢量化，但根据我的代码结合了多个操作。我想知道编写这个循环的最佳方法是什么，自动矢量化器会找到正确的 addat 操作匹配。

向量结构为：

struct V {
  static constexpr int length = 32;
  unsigned short data[32];
};

使用的编译器是 clang 3.8，代码是为 AVX2 Haswell x86-64 架构编译的。

4

0 回答 0