SSE实现Pade正弦/余弦函数的循环计数器控制问题

问题背景 #

我正在将J. M. ain't a mathematician的Mathematica Pade正弦代码转换为支持SSE2的C++代码，非SSE版本已完成，但作为SSE新手，在while循环中控制元素计算时遇到了问题：

• 非SSE版本通过单个变量k控制双角恒等式的循环次数，但SSE实现中每个向量元素需要独立的k值，该值由重复减半循环生成。例如输入向量{0.5, 1.01, -0.23, 0.99}、f=0.25时，减半后得到kvec={0,2,0,1}，即一个元素需计算2次循环，一个元素计算1次，其余元素跳过循环。
• 当前按最大k值执行循环，每次循环用multiplier向量控制参与计算的元素：第1次循环multiplier={0,1,0,1}，第2次{0,1,0,0}，第3次{0,0,0,0}。但现有求和方式结果错误，使用_mm_max_ps时正数结果正常，但负数会出现**-0**问题。

核心矛盾 #

1. SSE向量中各元素的循环次数独立，按最大次数循环时需要精准控制单个元素是否参与迭代更新。
2. _mm_max_ps处理负数时会保留负零，导致最终计算结果不符合预期。

解决方案 #

1. 用掩码替代_mm_max_ps实现条件更新 #

放弃_mm_max_ps的隐式条件判断，改用位掩码精准控制哪些元素参与当前循环的增量计算：

• 生成掩码：将multiplier中值为1的位置设为全1（0xFFFFFFFF），值为0的位置设为全0（0x00000000）。
• 过滤增量：用_mm_and_ps将当前循环计算出的增量值与掩码相乘，仅保留需要更新的元素的增量。
• 更新结果：用_mm_add_ps将过滤后的增量加到结果上，不需要更新的元素因增量为0，结果保持不变。

代码示例：

// 生成参与计算的元素掩码
__m128 mask = _mm_cmpgt_ps(multiplier, _mm_setzero_ps());
// 计算当前循环的增量值（根据你的Pade正弦逻辑实现）
__m128 delta = ...;
// 仅保留需要更新的元素的增量
delta = _mm_and_ps(delta, mask);
// 更新结果向量
result = _mm_add_ps(result, delta);

2. 消除负零问题 #

计算完成后，通过以下方式将负零转换为正零：

方式一：按位或正零

利用正零与负零的按位或结果为正零的特性，直接消除负零：

__m128 zero = _mm_setzero_ps();
result = _mm_or_ps(result, zero);

方式二：检测并替换负零

先检测向量中的负零（符号位为1，其余位为0），再将其替换为正零：

__m128 zero = _mm_setzero_ps();
// 检测负零：等于零且符号位为负
__m128 is_neg_zero = _mm_and_ps(_mm_cmpeq_ps(result, zero), _mm_cmpgt_ps(zero, result));
// 用正零替换负零
result = _mm_blendv_ps(result, zero, is_neg_zero);

内容的提问来源于stack exchange，提问作者Juha P