mlir-hlo/lib/Dialect/mhlo
Benjamin Kramer 4d435a817e [mhlo:linalg] Add support for lowering mhlo.concatenate to Linalg ops.
This uses a indexed linalg.generic, which is rather awkward standalone but
allows fusing into the output of the concatenate and avoid to ever materialize
it in memory. I think this is the only way to get that with the current linalg
stack, fusion across a concatenate would require more infrastructure.

PiperOrigin-RevId: 369677652
2021-04-21 10:01:08 -07:00
..
IR [mhlo] Add a folder for mhlo.map which does nothing but return one of the arguments. 2021-04-19 14:36:08 -07:00
transforms [mhlo:linalg] Add support for lowering mhlo.concatenate to Linalg ops. 2021-04-21 10:01:08 -07:00
CMakeLists.txt Add CMake files and lit configurations, enough for `ninja check-mlir-hlo` to pass on all the tests 2020-08-07 22:14:34 -07:00