为何WGSL中的2Sum算法无法正常运行？

问题分析 #

你实现的Knuth版2Sum算法在WGSL中返回错误结果，核心原因是WGSL编译器的激进浮点优化破坏了算法依赖的精确舍入中间结果。

2Sum算法的本质是利用浮点运算的舍入特性，分离出加法操作中丢失的精度部分。但WGSL默认会对浮点代码进行代数化简优化（比如将(a + b) - a直接替换为b），这种优化忽略了浮点运算的舍入误差，导致算法完全失效。

以你的测试用例twoSum(1e10, 1e-10)为例：

1. x = a + b在f32中会舍入为1e10（因为1e-10的量级远小于1e10，超出f32的尾数精度范围）
2. 若编译器未优化，bv = x - a应得到0，后续br = b - bv会得到1e-10，最终ar + br应为1e-10
3. 但WGSL的优化会直接推导ar + br的代数结果为0，或跳过关键的中间运算步骤，导致返回0

而C++代码未开启快速数学优化（默认行为），编译器保留了浮点运算的精确步骤；Python使用的float64精度更高，a + b不会完全丢失b的信息，因此两者都能正确运行。

解决方案 #

在WGSL中，通过volatile关键字强制编译器严格执行每一步浮点运算，禁止代数化简和结果推导：

struct float2 {
  x: f32,
  y: f32
}

fn twoSum(a: f32, b: f32) -> float2 {
  volatile var x = a + b;
  volatile var bv = x - a;
  volatile var av = x - bv;
  volatile var br = b - bv;
  volatile var ar = a - av;
  return float2(x, ar + br);
}

volatile变量的运算会被编译器视为"可能有外部副作用"，必须严格按照代码顺序执行，保留每一步的舍入结果，这正是2Sum算法所需的精确执行环境。

验证说明 #

修改后重新测试twoSum(1e10, 1e-10).y，应返回1e-10的近似值（受f32精度限制，可能存在微小误差）。若仍有问题，需检查WGSL编译选项是否强制开启了极端优化（如fast-math），确保编译时保留浮点运算的精确性。

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