Update DLTI tests

Author: rolfmorel

mlir/test/Target/LLVMIR/target-to-target-features-dlti-query.mlir

@@ -1,6 +1,6 @@
 // REQUIRES: target=x86{{.*}}
 
-// RUN: mlir-opt -transform-interpreter -split-input-file %s | FileCheck %s
+// RUN: mlir-opt -transform-interpreter -split-input-file %s --verify-diagnostics
 
 // Check that processor features, like AVX, are appropriated derived and queryable.
 
@@ -25,13 +25,39 @@ module attributes {transform.with_named_sequence} {
   }
 }
 
+
+// -----
+
+// Check that newer processor features, like AMX, are appropriated derived and queryable.
+
+// expected-remark @+2 {{attr associated to ["features", "+amx-bf16"] = unit}}
+// expected-remark @below {{attr associated to ["features", "amx-bf16"] = true}}
+module attributes { llvm.target = #llvm.target<triple = "x86_64-unknown-linux",
+                                               chip = "sapphirerapids">,
+                    test.dl_spec = #dlti.dl_spec<index = 32> } {
+  func.func private @f()
+}
+
+module attributes {transform.with_named_sequence} {
+  transform.named_sequence @__transform_main(%arg: !transform.any_op) {
+    %funcs = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
+    %module = transform.get_parent_op %funcs : (!transform.any_op) -> !transform.any_op
+    %mod = transform.apply_registered_pass "llvm-target-to-target-features" to %module : (!transform.any_op) -> !transform.any_op
+    %plus_avx = transform.dlti.query ["features", "+amx-bf16"] at %mod : (!transform.any_op) -> !transform.any_param
+    transform.debug.emit_param_as_remark %plus_avx, "attr associated to [\"features\", \"+amx-bf16\"] =" at %mod : !transform.any_param, !transform.any_op
+    %avx = transform.dlti.query ["features", "amx-bf16"] at %mod : (!transform.any_op) -> !transform.any_param
+    transform.debug.emit_param_as_remark %avx, "attr associated to [\"features\", \"amx-bf16\"] =" at %mod : !transform.any_param, !transform.any_op
+    transform.yield
+  }
+}
+
 // -----
 
-// Check that features that a processor does not have, AVX512f in this case,
+// Check that features that a processor does not have, AMX in this case,
 // aren't derived and hence that querying for them will fail.
 
 // expected-error @+2 {{target op of failed DLTI query}}
-// expected-note @below {{key "+avx512f" has no DLTI-mapping per attr: #llvm.target_features<[}}
+// expected-note @below {{key "+amx-bf16" has no DLTI-mapping per attr: #llvm.target_features}}
 module attributes { llvm.target = #llvm.target<triple = "x86_64-unknown-linux",
                                                chip = "skylake">,
                     test.dl_spec = #dlti.dl_spec<index = 32> } {
@@ -44,7 +70,7 @@ module attributes {transform.with_named_sequence} {
     %module = transform.get_parent_op %funcs : (!transform.any_op) -> !transform.any_op
     %mod = transform.apply_registered_pass "llvm-target-to-target-features" to %module : (!transform.any_op) -> !transform.any_op
     // expected-error @below {{'transform.dlti.query' op failed to apply}}
-    %param = transform.dlti.query ["features", "+avx512f"] at %mod : (!transform.any_op) -> !transform.any_param
+    %param = transform.dlti.query ["features", "+amx-bf16"] at %mod : (!transform.any_op) -> !transform.any_param
     transform.yield
   }
 }