diff options
Diffstat (limited to 'mlir/lib/Dialect/Tensor/IR/TensorOps.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Tensor/IR/TensorOps.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp index b21e89a..8298cf1 100644 --- a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp +++ b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp @@ -2663,8 +2663,8 @@ struct InsertSliceOpSourceCastInserter final if (!hasValidSizesOffsets(newSrcShape)) return failure(); - RankedTensorType newSrcType = - RankedTensorType::get(newSrcShape, srcType.getElementType()); + RankedTensorType newSrcType = RankedTensorType::get( + newSrcShape, srcType.getElementType(), srcType.getEncoding()); if (srcType == newSrcType || !preservesStaticInformation(srcType, newSrcType) || !tensor::CastOp::areCastCompatible(srcType, newSrcType)) |