iree icon indicating copy to clipboard operation
iree copied to clipboard

[Codegen] Support dynamic dimensions in collapse_shape into interface store folding

Open jtuyls opened this issue 6 months ago • 0 comments

Towards: https://github.com/iree-org/iree/issues/20835

Supporting dynamic dimensions in the tensor.collapse_shape into iree_tensor_ext.dispatch.tensor.store folding is needed to support pad encodings in e2e Llama3 as the remaining collapses will otherwise result in linalg.copy operations on workspace and global memory (which results in a compilation failure).

jtuyls avatar Jun 18 '25 16:06 jtuyls