[PATCH] D80347: [mlir][Linalg] Add support for fusion between indexed_generic ops and generic ops on tensors.

[Mahesh Ravishankar via Phabricator via llvm-commits]

mravishankar requested changes to this revision.
mravishankar added inline comments.
This revision now requires changes to proceed.


================
Comment at: mlir/lib/Dialect/Linalg/Transforms/Fusion.cpp:656
+    // consumer's operand.
+    unsigned numProducerIndices =
+        isa<IndexedGenericOp>(producer.getOperation()) ? nloops : 0;
----------------
It seems like the logic used for generating the fused region for the indexed generic op would work for generic op as well. The numProducerIndices and numConsumerIndices would just remain the same. Can we combine these two methods?


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D80347/new/

https://reviews.llvm.org/D80347