[Mlir-commits] [mlir] [MLIR][DLTI] Enable types as keys in DLTI-query utils (PR #105995)

[Adam Siemieniuk]

================
@@ -33,7 +33,16 @@ void transform::QueryOp::getEffects(
 DiagnosedSilenceableFailure transform::QueryOp::applyToOne(
     transform::TransformRewriter &rewriter, Operation *target,
     transform::ApplyToEachResultList &results, TransformState &state) {
-  auto keys = SmallVector<StringAttr>(getKeys().getAsRange<StringAttr>());
+  auto keys = SmallVector<DataLayoutEntryKey>();
----------------
adam-smnk wrote:

```suggestion
  SmallVector<DataLayoutEntryKey> keys;
```

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