[PATCH] D73796: [mlir][Linalg] Adding support for linalg_matmul with tensors.

Nicolas Vasilache via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Fri Jan 31 11:49:40 PST 2020


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

This revision provides 2 versions of matmul with tensors to account for the differences in buffer vs value semantics:

1. `C(i, j) = sum_{r_k} A(i, r_k) * B(r_k, j)`
2. `D(i, j) = C(i, j) + sum_{r_k} A(i, r_k) * B(r_k, j)`


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D73796

Files:
  mlir/include/mlir/Dialect/Linalg/EDSC/Builders.h
  mlir/lib/Dialect/Linalg/EDSC/Builders.cpp
  mlir/test/EDSC/builder-api-test.cpp

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D73796.241783.patch
Type: text/x-patch
Size: 9491 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20200131/19af14f5/attachment-0001.bin>


More information about the llvm-commits mailing list