iree
iree copied to clipboard
Materialize inserted indices for mhlo.scatter
mhlo.scatter supports insertedd dimensions at any index length-1 or indexed into. Linalg's scatter requires that indices be explicitly defined (excluding optionally inserting the indexed dims). Adds an mhlo-to-mhlo rewriter that materializes the cases where the inserted indices occur after index dims.