fengxie wrote: > Can you add a test? (the example you have in your description looks like a good candidate for a test?) Just added a test in `mlir/test/Dialect/GPU/outlining.mlir` https://github.com/llvm/llvm-project/pull/72074