[PATCH] D158586: [mlir][ArmSME] Lower vector.broadcast to ArmSME

[Cullen Rhodes via Phabricator via llvm-commits]

c-rhodes created this revision.
c-rhodes added reviewers: awarzynski, benmxwl-arm.
c-rhodes added a project: MLIR.
Herald added subscribers: bviyer, Moerafaat, zero9178, bzcheeseman, sunshaoce, sdasgup3, wenzhicui, wrengr, cota, teijeong, rdzhabarov, tatianashp, msifontes, jurahul, Kayjukh, grosul1, Joonsoo, liufengdb, aartbik, mgester, arpith-jacob, antiagainst, shauheen, rriddle, mehdi_amini, kristof.beyls.
Herald added a reviewer: aartbik.
Herald added a project: All.
c-rhodes requested review of this revision.
Herald added a reviewer: nicolasvasilache.
Herald added subscribers: wangpc, alextsao1999, limo1996, stephenneuendorffer, nicolasvasilache.
Herald added a reviewer: dcaballe.

This adds support for lowering vector.broadcast ops to SME, if the
source is either a scalar, 0-d vector, or 1-d vector, and the result a
2-d scalable vector that aligns with SME tiles.

This follows on from D157005 <https://reviews.llvm.org/D157005> which introduced a vector to tile slice op
that moves a 1-d scalable vector to a slice of a 2-d scalable vector
(tile). The lowering from vector.broadcast is similar, a couple of
helper functions are added to prevent duplication.

Lowering of vector.broadcast contributes towards a path from linalg.fill
to SME.

Depends on D157005 <https://reviews.llvm.org/D157005>


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D158586

Files:
  mlir/lib/Conversion/VectorToArmSME/VectorToArmSME.cpp
  mlir/test/Dialect/ArmSME/vector-ops-to-sme.mlir

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D158586.552614.patch
Type: text/x-patch
Size: 8443 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20230823/82b5fce4/attachment.bin>