[Indexing] Add transform from scatter/gather to insert_slice/extract_slice

iree-org / iree-llvm-sandbox

A sandbox for quick iteration and experimentation on projects related to IREE, MLIR, and LLVM

Apache License 2.0

54 stars 31 forks source link

[Indexing] Add transform from scatter/gather to insert_slice/extract_slice #742

Closed makslevental closed 8 months ago

makslevental commented 1 year ago

Depends on https://github.com/iree-org/iree-llvm-sandbox/pull/753

This PR adds a lowering from the simplest scatter/gather case (regular start, stop, stop for all indices) to tensor.insert_slice/tensor.extract_slice.