iree icon indicating copy to clipboard operation
iree copied to clipboard

Materialize inserted indices for mhlo.scatter

Open rsuderman opened this issue 3 years ago • 0 comments

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.

rsuderman avatar Sep 17 '22 00:09 rsuderman