[mlir][Vector] Don't fully unroll transfer_writes of n-D scalable vectors (#71924)
It is not possible to unroll a scalable vector at compile time. This currently prevents transfer_writes from being lowered to arm_sme.tile_writes (downstream).
This commit is contained in:
@@ -1218,6 +1218,11 @@ struct UnrollTransferWriteConversion
|
||||
|
||||
auto vec = getDataVector(xferOp);
|
||||
auto xferVecType = xferOp.getVectorType();
|
||||
if (xferVecType.getScalableDims()[0]) {
|
||||
// Cannot unroll a scalable dimension at compile time.
|
||||
return failure();
|
||||
}
|
||||
|
||||
int64_t dimSize = xferVecType.getShape()[0];
|
||||
Value source = xferOp.getSource(); // memref or tensor to be written to.
|
||||
auto sourceType = isTensorOp(xferOp) ? xferOp.getShapedType() : Type();
|
||||
|
||||
Reference in New Issue
Block a user