We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 18b4cf4 commit f2e1ec8Copy full SHA for f2e1ec8
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
@@ -2405,7 +2405,10 @@ BroadcastableToResult mlir::vector::isBroadcastableTo(
2405
bool srcDimScalableFlag = srcVectorType.getScalableDims()[dimIdx];
2406
bool dstDimScalableFlag = dstVectorType.getScalableDims()[lead + dimIdx];
2407
if ((srcDim == 1 && srcDimScalableFlag && dstDim != 1) ||
2408
- (srcDimScalableFlag != dstDimScalableFlag))
+ // 1 -> [N] is fine, everything else should be rejected when mixing
2409
+ // fixed-width and scalable dims
2410
+ (srcDimScalableFlag != dstDimScalableFlag &&
2411
+ (srcDim != 1 || srcDimScalableFlag)))
2412
foundMismatchingDims = true;
2413
2414
if (foundMismatchingDims) {
0 commit comments