[PATCH] D76456: [mlir][Linalg][RFC] Add basic plumbing for "named ops" in Linalg

Nicolas Vasilache via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Thu Mar 19 14:50:02 PDT 2020


nicolasvasilache created this revision.
Herald added subscribers: llvm-commits, Joonsoo, liufengdb, aartbik, lucyrfox, mgester, arpith-jacob, antiagainst, shauheen, burmako, jpienaar, rriddle, mehdi_amini, mgorny.
Herald added a reviewer: rriddle.
Herald added a project: LLVM.
nicolasvasilache updated this revision to Diff 251480.
nicolasvasilache added a comment.

Remove duplicate assemblyFormat


This revision sketches how Linalg ops can be defined declaratively using a language like Tensor Comprehensions.

For now this does not invest more in the infrastructure than the bare minimum to demonstrate end-to-end op generation with Tablegen. At the moment the language is left unspecified and we only do basic string matching to a predefined C++ implementation.

In the longer term, a large fraction of ops can be written declaratively as custom ops with a single mathematical source of truth that resembles:

  "C(i,j) +=! A(i,k) * B(k,j)"


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D76456

Files:
  mlir/include/mlir/Dialect/Linalg/IR/CMakeLists.txt
  mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
  mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h
  mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
  mlir/lib/Dialect/Linalg/Transforms/LinalgToLoops.cpp
  mlir/lib/IR/AffineMap.cpp
  mlir/test/Dialect/Linalg/roundtrip.mlir
  mlir/tools/mlir-tblgen/CMakeLists.txt
  mlir/tools/mlir-tblgen/LinalgNamedOpsGen.cpp

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D76456.251480.patch
Type: text/x-patch
Size: 14922 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200319/0a4f1aa0/attachment.bin>


More information about the llvm-commits mailing list