forked from OSchip/llvm-project
[mlir] Disallow tensor types in VectorToSCF
Support for tensor types can be added if needed. Differential Revision: https://reviews.llvm.org/D102749
This commit is contained in:
parent
884a6291f0
commit
8fb4897934
mlir/lib/Conversion/VectorToSCF
|
@ -485,6 +485,8 @@ LogicalResult checkPrepareXferOp(OpTy xferOp,
|
||||||
return failure();
|
return failure();
|
||||||
if (xferOp.getVectorType().getRank() <= options.targetRank)
|
if (xferOp.getVectorType().getRank() <= options.targetRank)
|
||||||
return failure();
|
return failure();
|
||||||
|
if (xferOp.getShapedType().template isa<RankedTensorType>())
|
||||||
|
return failure();
|
||||||
return success();
|
return success();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -802,6 +804,8 @@ struct UnrollTransferReadConversion
|
||||||
PatternRewriter &rewriter) const override {
|
PatternRewriter &rewriter) const override {
|
||||||
if (xferOp.getVectorType().getRank() <= options.targetRank)
|
if (xferOp.getVectorType().getRank() <= options.targetRank)
|
||||||
return failure();
|
return failure();
|
||||||
|
if (xferOp.getShapedType().template isa<RankedTensorType>())
|
||||||
|
return failure();
|
||||||
|
|
||||||
auto insertOp = getInsertOp(xferOp);
|
auto insertOp = getInsertOp(xferOp);
|
||||||
auto vec = getResultVector(xferOp, rewriter);
|
auto vec = getResultVector(xferOp, rewriter);
|
||||||
|
@ -918,6 +922,8 @@ struct UnrollTransferWriteConversion
|
||||||
PatternRewriter &rewriter) const override {
|
PatternRewriter &rewriter) const override {
|
||||||
if (xferOp.getVectorType().getRank() <= options.targetRank)
|
if (xferOp.getVectorType().getRank() <= options.targetRank)
|
||||||
return failure();
|
return failure();
|
||||||
|
if (xferOp.getShapedType().template isa<RankedTensorType>())
|
||||||
|
return failure();
|
||||||
|
|
||||||
auto vec = getDataVector(xferOp);
|
auto vec = getDataVector(xferOp);
|
||||||
auto xferVecType = xferOp.getVectorType();
|
auto xferVecType = xferOp.getVectorType();
|
||||||
|
|
Loading…
Reference in New Issue