The fusion for tensor_reshape is embedding the information to indexing maps,
thus the exising pattenr also works for indexed_generic ops.
Depends On D80347
Differential D80348
[mlir][Linalg] Add support for fusion between indexed_generic ops and tensor_reshape ops hanchung on May 20 2020, 5:48 PM. Authored by
Details The fusion for tensor_reshape is embedding the information to indexing maps, Depends On D80347
Diff Detail
Event TimelineComment Actions Good to know that this minimal change works as expected (I hoped it would). Thanks for adding tests. Comment Actions This is a bit tricky, we still need the change to deduce the type of the op to create (whether this is GenericOp or IndexedGenericOp). Comment Actions
I agree. This is more evidence for me that the indexed-generic and generic should really become op-interfaces after the named ops are replaced with generated ops in Linalg. @nicolasvasilache what do you think. I can start a discussion on discourse if thats a better forum. |