[Mlir-commits] [mlir] [mlir][dlti] Fix query keys preallocation (PR #127786)

[llvmlistbot at llvm.org]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlir-dlti

Author: Adam Siemieniuk (adam-smnk)

<details>
<summary>Changes</summary>

Fixes upfront space allocation after #<!-- -->126716

---
Full diff: https://github.com/llvm/llvm-project/pull/127786.diff


1 Files Affected:

- (modified) mlir/lib/Dialect/DLTI/DLTI.cpp (+2-1) 


``````````diff
diff --git a/mlir/lib/Dialect/DLTI/DLTI.cpp b/mlir/lib/Dialect/DLTI/DLTI.cpp
index b057554c40d8c..70e05cb4cb383 100644
--- a/mlir/lib/Dialect/DLTI/DLTI.cpp
+++ b/mlir/lib/Dialect/DLTI/DLTI.cpp
@@ -571,7 +571,8 @@ FailureOr<Attribute> dlti::query(Operation *op, ArrayRef<StringRef> keys,
     return failure();
 
   MLIRContext *ctx = op->getContext();
-  SmallVector<DataLayoutEntryKey> entryKeys(keys.size());
+  SmallVector<DataLayoutEntryKey> entryKeys;
+  entryKeys.reserve(keys.size());
   for (StringRef key : keys)
     entryKeys.push_back(StringAttr::get(ctx, key));
 

``````````

</details>


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