[Mlir-commits] [mlir] [mlir] Fix use-after-free bugs in {RankedTensorType|VectorType}::Builder (PR #68969)

Benjamin Maxwell llvmlistbot at llvm.org
Mon Oct 16 01:44:58 PDT 2023


================
@@ -277,7 +277,7 @@ class RankedTensorType::Builder {
     if (storage.empty())
       storage.append(shape.begin(), shape.end());
     storage.erase(storage.begin() + pos);
-    shape = {storage.data(), storage.size()};
+    shape = {};
----------------
MacDue wrote:

Oh sorry, I didn't notice that assert. That should be updated to `getShape()`, I would guess it's not triggered right now as noting does multiple `.dropDims()` in a row. The invalid shape bug still exists with a single `dropDim()` call.

https://github.com/llvm/llvm-project/pull/68969


More information about the Mlir-commits mailing list