diff --git a/SUBMODULE_VERSIONS.txt b/SUBMODULE_VERSIONS.txt index a2294099e233..2b0179e2a0b1 100644 --- a/SUBMODULE_VERSIONS.txt +++ b/SUBMODULE_VERSIONS.txt @@ -5,7 +5,7 @@ b1fbd33c06cdb0024c67733c6fdec2009d17b384 third_party/googletest 88b845dee001723c4a0db1fe5477de735b6d3bb0 third_party/liburing 189e771009a640214e08e855830ae6f15a83c655 third_party/llvm-bazel -20d5c42e0ef5d252b434bcb610b04f1cb79fe771 third_party/llvm-project +482283042f795ecc27838a3b2f76b5494991401c third_party/llvm-project dde739ffd00a6fa99175cf3c0f28e4b763dc6f5f third_party/mlir-emitc cbef26c6a8f1e4be3f4cfb902db992c45e93b7a6 third_party/mlir-hlo 2b2bd45bbf9be04fd22ece5cc1f54679202e9257 third_party/pffft diff --git a/iree/compiler/Conversion/Common/LinalgBufferizePass.cpp b/iree/compiler/Conversion/Common/LinalgBufferizePass.cpp index 10909b4e89cd..b40e6e798c99 100644 --- a/iree/compiler/Conversion/Common/LinalgBufferizePass.cpp +++ b/iree/compiler/Conversion/Common/LinalgBufferizePass.cpp @@ -329,12 +329,14 @@ static LogicalResult convertTensorReshapeOp( resultTensorType, {}, inputBufferType.getMemorySpaceAsInt()); Value bufferReshape = b.create( loc, reshapeResultType, reshapeSrc, op.reassociation()); - auto allocationDynamicSizes = linalg::getReshapeOutputShapeFromInputShape( - b, loc, inputBuffer, resultTensorType.getShape(), - op.getReassociationMaps()); + SmallVector> reshapeResultShape; + if (failed(op.reifyReturnTypeShapesPerResultDim(b, reshapeResultShape)) || + reshapeResultShape.size() != 1) { + return op.emitError("failed to get shape of result"); + } return createAliasingBufferOrAllocationForResult( b, loc, allocationFn, srcTensor, bufferReshape, resultTensor, - allocationDynamicSizes, bvm); + reshapeResultShape[0], bvm); } static SmallVector extractFromI64ArrayAttr(ArrayAttr attr) { diff --git a/third_party/llvm-project b/third_party/llvm-project index 20d5c42e0ef5..482283042f79 160000 --- a/third_party/llvm-project +++ b/third_party/llvm-project @@ -1 +1 @@ -Subproject commit 20d5c42e0ef5d252b434bcb610b04f1cb79fe771 +Subproject commit 482283042f795ecc27838a3b2f76b5494991401c