dcaballe wrote: Thanks! > Would be nice to leave a TODO somewhere to handle this in folds/cannon patterns, e.g., extract, extract_strided_slice, etc. I added this to the ops' doc. I already have some of this implemented. https://github.com/llvm/llvm-project/pull/122188