Address comments

Author: adam-smnk

mlir/include/mlir/Dialect/DLTI/DLTI.h

@@ -29,9 +29,9 @@ namespace dlti {
 FailureOr<Attribute> query(Operation *op, ArrayRef<DataLayoutEntryKey> keys,
                            bool emitError = false);
 
-/// Perform a DLTI-query at `op` using string `keys` as DLTI entry keys,
-/// recursively querying on query interface-implementing attrs, starting from
-/// attr obtained from `op`.
+/// Perform a DLTI-query at `op` using each string in `keys` as a separate DLTI
+/// entry key, recursively querying on query interface-implementing attrs,
+/// starting from attr obtained from `op`.
 FailureOr<Attribute> query(Operation *op, ArrayRef<StringRef> keys,
                            bool emitError = false);
 } // namespace dlti

mlir/lib/Dialect/DLTI/DLTI.cpp

@@ -571,7 +571,7 @@ FailureOr<Attribute> dlti::query(Operation *op, ArrayRef<StringRef> keys,
     return failure();
 
   MLIRContext *ctx = op->getContext();
-  SmallVector<DataLayoutEntryKey> entryKeys;
+  SmallVector<DataLayoutEntryKey> entryKeys(keys.size());
   for (StringRef key : keys)
     entryKeys.push_back(StringAttr::get(ctx, key));