Binaryen项目中常量传播与分支优化的深入分析

背景介绍

Binaryen是一个WebAssembly编译器工具链基础设施项目，它提供了多种优化WebAssembly代码的功能。在最新开发过程中，我们发现了一个关于常量传播与分支优化的有趣案例，值得深入探讨。

问题现象

我们观察到一个特定的WebAssembly模块，在使用不同优化级别时表现出不同的行为：

(module
  (import "External" "external_function" (func $external_function))
  (func $_start 
    i32.const 0
    i32.const 0
    call $foo)
  (func $foo (param $0 i32) (param $2 i32) 
    block ;; label = @1
      i32.const 0
      i32.load
      drop
      i32.const 1
      i32.load
      local.set $2
      i32.const 1
      local.set $0
      local.get $2
      br_if 0 (;@1;)
      i32.const 1
      local.set $0
      local.get $0
      br_if 0 (;@1;)
      i32.const 0
      local.set $0
    end
    block ;; label = @1
      local.get $0
      br_if 0 (;@1;)
      call $external_function
    end)
  (memory $0 258 258)
  (export "_start" (func $_start)))

当使用-all -O2优化时，Binaryen能够正确推导出第二个块中的分支条件为真，但在-all -O3优化级别下却无法实现这一优化。

优化结果分析

经过-all -O3优化后，代码变为：

(if
 (i32.eqz
  (block $block (result i32)
   (drop
    (i32.load (i32.const 0))
   )
   (drop
    (br_if $block
     (i32.const 1)
     (i32.load (i32.const 1))
    )
   )
   (i32.const 1)
  )
 )
 (then (call $external_function))
)

关键问题在于常量值1被"埋藏"在块结构中，无法被i32.eqz识别。这与之前发现的其他问题类似，但有一个重要区别：这些块具有名称，阻止了它们成为"fallthrough"块。

技术原理

在Binaryen的优化过程中，getImmediateFallthroughPtr函数负责处理块的fallthrough情况。当块有名称时，它们不会被标记为fallthrough，这影响了优化效果。

核心优化机会在于：当一个块以br_if和一个值结束时，如果br_if的值与最终值相同，并且该值没有副作用，我们可以简化结构。这种优化应该放在RemoveUnusedBrs中，该模块已经处理了类似的块尾优化。

解决方案

我们可以实现以下转换：

(block $block (result i32)
 ...
 (drop
  (br_if $block
   (value)
   (condition)
  )
 )
 (value)
)

=>

(block $block (result i32)
 ...
 (drop
  (condition)
 )
 (value)
)

这种转换保留了语义，同时为后续的常量传播优化创造了条件。

总结

这个案例展示了WebAssembly优化过程中的一些微妙之处。块命名、控制流分析和常量传播之间的相互作用可能导致优化机会的丢失。通过识别特定模式并实施有针对性的优化，我们可以提高Binaryen的优化能力，特别是在处理带有命名块的代码时。这种优化不仅适用于当前案例，也可以推广到其他类似的控制流模式中。