mirror of https://github.com/llvm/torch-mlir
ef25428fe3
This also adds a small pass to clean up the `dim` ops that linalg introduces. For now, it only has a trivial pattern that looks for a `tcp.alloc_memref(%shape)` op to get the shape as we currently have an invariant that all memrefs are the result of such ops. But eventually this will need to look through view ops and any other shape-ish stuff that linalg introduces as it lowers to loops, along with any slicing ops introduced by buffer allocation. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| E2E.h | ||
| Passes.td | ||