这是一个相当直接的解决方案,它只是用SSE实现标量代码而不需要进一步优化。它可能会更有效率,例如通过利用当wZ = -1.0时结果为0.5的事实,或者甚至可以通过仅仅进行除法,然后在事实之后将INF
s转换为0.5来实现。
对于SSE4与pre-SSE4,我已经有#ifdef
d了,因为SSE4有一个“混合”指令,这个指令可能更有效一些屏蔽和选择值所需的三个预先SSE4指令。
#include <emmintrin.h>
#ifdef __SSE4_1__
#include <smmintrin.h>
#endif
#include <stdio.h>
int main(void)
{
const __m128 vk1 = _mm_set1_ps(1.0f); // useful constants
const __m128 vk0 = _mm_set1_ps(0.0f);
__m128 wZ, d, d0, d1, vcmp;
#ifndef __SSE4_1__ // pre-SSE4 implementation
__m128 d0_masked, d1_masked;
#endif
wZ = _mm_set_ps(-1.0f, 0.0f, 1.0f, 2.0f); // test inputs
d0 = _mm_add_ps(vk1, wZ); // d0 = 1.0 - wZ
d1 = _mm_sub_ps(vk1, wZ); // d1 = 1.0 + wZ
vcmp = _mm_cmpneq_ps(d1, vk0); // test for d1 != 0.0, i.e. wZ != -1.0
#ifdef __SSE4_1__ // SSE4 implementation
d = _mm_blendv_ps(d0, d1, vcmp);
#else // pre-SSE4 implementation
d0_masked = _mm_andnot_ps(vcmp, d0);
d1_masked = _mm_and_ps(vcmp, d1);
d = _mm_or_ps(d0_masked, d1_masked); // d = wZ == -1.0 ? 1.0/(1.0 - wZ) : 1.0/(1.0 + wZ)
#endif
d = _mm_div_ps(vk1, d);
printf("wZ = %vf\n", wZ);
printf("d = %vf\n", d);
return 0;
}
除以零有什么问题? – Pubby
除了明显的问题之外,它还会通过在算法的其余部分为Nz = -1创建不一致性来破坏结果。 – cubiclewar