Skip to content

Commit 66ecff4

Browse files
committed
updates
1 parent 3e2e4b5 commit 66ecff4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

mlir/lib/Dialect/Vector/Transforms/VectorEmulateNarrowType.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -380,8 +380,9 @@ static void rmwStore(OpBuilder &builder, Location loc,
380380
origVecValue);
381381

382382
// Construct the final masked value and yield it.
383-
Value maskedValue = selectAndCast(builder, loc, oneElemVecType, mask,
384-
origVecValue, valueToStore);
383+
Value maskedValue =
384+
downcastSelectAndUpcast(builder, loc, valueToStore.getType(),
385+
oneElemVecType, mask, valueToStore, origVecValue);
385386
builder.create<vector::StoreOp>(loc, maskedValue, linearizedMemref,
386387
linearizedIndex);
387388
}

0 commit comments

Comments
 (0)