diff options
| author | Cullen Rhodes <cullen.rhodes@arm.com> | 2023-07-18 08:29:53 +0000 |
|---|---|---|
| committer | Cullen Rhodes <cullen.rhodes@arm.com> | 2023-07-18 08:46:40 +0000 |
| commit | fb54fec7263ab420e9e02e1ea2b66403cf655a33 (patch) | |
| tree | 2f4d965725c47da93b766a6a6f02c5ea1d2c3805 /llvm/lib/IR/Function.cpp | |
| parent | 0db5d8e12384d780d2ded6b972233dc8b1632560 (diff) | |
| download | llvm-fb54fec7263ab420e9e02e1ea2b66403cf655a33.zip llvm-fb54fec7263ab420e9e02e1ea2b66403cf655a33.tar.gz llvm-fb54fec7263ab420e9e02e1ea2b66403cf655a33.tar.bz2 | |
[mlir][ArmSME] Implement tile allocation
This patch adds a pass '-allocate-sme-tiles' to the ArmSME dialect that
implements allocation of SME ZA tiles.
It does this at the 'func.func' op level by replacing
'arm_sme.get_tile_id' ops with 'arith.constant' ops that represent the
tile number. The tiles in use in a given function are tracked by an
integer function attribute 'arm_sme.tiles_in_use' that is a 16-bit tile
mask with a bit for each 128-bit element tile (ZA0.Q-ZA15.Q), the
smallest ZA tile granule. This is initialized on the first
'arm_sme.get_tile_id' rewrite and updated on each subsequent rewrite.
Mixing of different element tile types is supported.
Section B2.3.2 of the SME spec [1] describes how the 128-bit element
tiles overlap with other element tiles.
Depends on D154941
[1] https://developer.arm.com/documentation/ddi0616/aa
Reviewed By: awarzynski
Differential Revision: https://reviews.llvm.org/D154955
Diffstat (limited to 'llvm/lib/IR/Function.cpp')
0 files changed, 0 insertions, 0 deletions