nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
69 stars 30 forks source link

[aievec] ShiftOp folder #871

Closed newling closed 2 weeks ago

newling commented 3 weeks ago

ShiftOp has 2 64 byte vector operands, and 1 64 byte vector result. The result is a concatenation of the N upper bytes from the first operand, and the 64 - N lower bytes from the second operand. This folder considers the special cases where N = 0 and N = 64, when it is possible to just return one of the operands directly.

https://www.xilinx.com/htmldocs/xilinx2024_1/aiengine_ml_intrinsics/intrinsics/group__intr__gpvectorop__shift.html#gaa83afc81114a5e0d3aebea1e973248bc