rsuderman wrote: > Can you add a test for this? Test added - I had to piggy back on top of the `tosa-to-linalg` pass as it appears related to args -> terminator, so defining that with a custom lowering ends with more work than would be preferred. https://github.com/llvm/llvm-project/pull/86991