[PATCH] D74172: [mlir][Linalg] Implement fusion of linalg.generic operation on tensors.

Mahesh Ravishankar via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Thu Feb 6 14:52:59 PST 2020


mravishankar created this revision.
Herald added subscribers: llvm-commits, Joonsoo, liufengdb, aartbik, lucyrfox, mgester, arpith-jacob, nicolasvasilache, antiagainst, shauheen, burmako, jpienaar, rriddle, mehdi_amini.
Herald added a reviewer: nicolasvasilache.
Herald added a project: LLVM.
mravishankar added a subscriber: asaadaldien.

The initial implementation of the fusion operation exposes a method to
fuse a consumer with its producer, when

- both the producer and consumer operate on tensors
- the producer has only a single result value
- the producer has only "parallel" iterator types

A new interface method hasTensorSemantics is added to verify that an
operation has all operands and results of type RankedTensorType.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D74172

Files:
  mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
  mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h
  mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
  mlir/lib/Dialect/Linalg/Transforms/Fusion.cpp
  mlir/test/Dialect/Linalg/fusion-tensor.mlir

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D74172.243016.patch
Type: text/x-patch
Size: 18200 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200206/c9a9b084/attachment.bin>


More information about the llvm-commits mailing list