5c1752e368
Unifies parsing and printing for DLTI attributes. Introduces a format of `#dlti.attr<key1 = val1, ..., keyN = valN>` syntax for all queryable DLTI attributes similar to that of the DictionaryAttr, while retaining support for specifying key-value pairs with `#dlti.dl_entry` (whether to retain this is TBD). As the new format does away with most of the boilerplate, it is much easier to parse for humans. This makes an especially big difference for nested attributes. Updates the DLTI-using tests and includes fixes for misc error checking/ error messages.
486 lines
21 KiB
MLIR
486 lines
21 KiB
MLIR
// RUN: mlir-opt -transform-interpreter -canonicalize -split-input-file -verify-diagnostics %s | FileCheck %s
|
|
|
|
// expected-remark @below {{attr associated to "test.id" = 42 : i32}}
|
|
module attributes { test.dlti = #dlti.map<"test.id" = 42 : i32> } {
|
|
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
|
|
%param = transform.dlti.query ["test.id"] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "attr associated to \"test.id\" =" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-remark @below {{i32 present in set : unit}}
|
|
module attributes { test.dlti = #dlti.map<i32 = unit> } {
|
|
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
|
|
%param = transform.dlti.query [i32] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "i32 present in set :" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-remark @below {{attr associated to i32's "width_in_bits" = 32 : i32}}
|
|
module attributes { test.dlti = #dlti.map<i32 = #dlti.map<"width_in_bits" = 32 : i32>> } {
|
|
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
|
|
%param = transform.dlti.query [i32,"width_in_bits"] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "attr associated to i32's \"width_in_bits\" =" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-remark @below {{width in bits of i32 = 32 : i64}}
|
|
// expected-remark @below {{width in bits of f64 = 64 : i64}}
|
|
module attributes { test.dlti = #dlti.map<"width_in_bits" = #dlti.map<i32 = 32, f64 = 64>> } {
|
|
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
|
|
%i32bits = transform.dlti.query ["width_in_bits",i32] at %module : (!transform.any_op) -> !transform.any_param
|
|
%f64bits = transform.dlti.query ["width_in_bits",f64] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %i32bits, "width in bits of i32 =" at %module : !transform.any_param, !transform.any_op
|
|
transform.debug.emit_param_as_remark %f64bits, "width in bits of f64 =" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-remark @below {{attr associated to "test.id" = 42 : i32}}
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32> } {
|
|
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
|
|
%param = transform.dlti.query ["test.id"] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "attr associated to \"test.id\" =" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32> } {
|
|
// expected-remark @below {{attr associated to "test.id" = 24 : i32}}
|
|
func.func private @f() attributes { test.dlti = #dlti.dl_spec<"test.id" = 24 : i32>}
|
|
}
|
|
|
|
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
|
|
%param = transform.dlti.query ["test.id"] at %funcs : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "attr associated to \"test.id\" =" at %funcs : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-remark @below {{attr associated to "test.id" = 42 : i32}}
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32> } {
|
|
func.func private @f() attributes { test.dlti = #dlti.dl_spec<"test.id" = 24 : i32> }
|
|
}
|
|
|
|
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
|
|
%param = transform.dlti.query ["test.id"] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "attr associated to \"test.id\" =" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32> } {
|
|
func.func @matmul_tensors(
|
|
%arg0: tensor<?x?xf32>, %arg1: tensor<?x?xf32>, %arg2: tensor<?x?xf32>)
|
|
-> tensor<?x?xf32> {
|
|
// expected-remark @below {{associated attr 42 : i32}}
|
|
%0 = linalg.matmul ins(%arg0, %arg1: tensor<?x?xf32>, tensor<?x?xf32>)
|
|
outs(%arg2: tensor<?x?xf32>)
|
|
-> tensor<?x?xf32>
|
|
return %0 : tensor<?x?xf32>
|
|
}
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%matmul = transform.structured.match ops{["linalg.matmul"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%param = transform.dlti.query ["test.id"] at %matmul : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "associated attr" at %matmul : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32> } {
|
|
func.func @matmul_tensors(
|
|
%arg0: tensor<?x?xf32>, %arg1: tensor<?x?xf32>, %arg2: tensor<?x?xf32>)
|
|
-> tensor<?x?xf32> attributes {test.dlti = #dlti.dl_spec<"test.id" = 24 : i32> } {
|
|
// expected-remark @below {{associated attr 24 : i32}}
|
|
%0 = linalg.matmul ins(%arg0, %arg1: tensor<?x?xf32>, tensor<?x?xf32>)
|
|
outs(%arg2: tensor<?x?xf32>)
|
|
-> tensor<?x?xf32>
|
|
return %0 : tensor<?x?xf32>
|
|
}
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%matmul = transform.structured.match ops{["linalg.matmul"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%param = transform.dlti.query ["test.id"] at %matmul : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "associated attr" at %matmul : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-remark @below {{associated attr 42 : i32}}
|
|
module attributes { test.dlti =
|
|
#dlti.target_system_spec<"CPU" =
|
|
#dlti.target_device_spec<"test.id" = 42 : i32>> } {
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%module = transform.get_parent_op %func : (!transform.any_op) -> !transform.any_op
|
|
%param = transform.dlti.query ["CPU","test.id"] at %module : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "associated attr" at %module : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.target_system_spec<"CPU" = #dlti.target_device_spec<"test.id" = 42 : i32>,
|
|
"GPU" = #dlti.target_device_spec<"test.id" = 43 : i32>> } {
|
|
// expected-remark @below {{associated attr 43 : i32}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%param = transform.dlti.query ["GPU","test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "associated attr" at %func : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.target_system_spec<"CPU" = #dlti.target_device_spec<"test.id" = 42 : i32>,
|
|
"GPU" = #dlti.target_device_spec<"test.id" = 43 : i32>> } {
|
|
// expected-remark @below {{associated attr 24 : i32}}
|
|
func.func private @f() attributes { test.dlti = #dlti.target_system_spec<"CPU" = #dlti.target_device_spec<"test.id" = 24 : i32>> }
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%param = transform.dlti.query ["CPU","test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.debug.emit_param_as_remark %param, "associated attr" at %func : !transform.any_param, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
"cache::L1::size_in_bytes" = 65536 : i32,
|
|
"cache::L1d::size_in_bytes" = 32768 : i32>> } {
|
|
// expected-remark @below {{L1::size_in_bytes 65536 : i32}}
|
|
// expected-remark @below {{L1d::size_in_bytes 32768 : i32}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%l1_size = transform.dlti.query ["CPU","cache::L1::size_in_bytes"] at %func : (!transform.any_op) -> !transform.param<i32>
|
|
%l1d_size = transform.dlti.query ["CPU","cache::L1d::size_in_bytes"] at %func : (!transform.any_op) -> !transform.param<i32>
|
|
transform.debug.emit_param_as_remark %l1_size, "L1::size_in_bytes" at %func : !transform.param<i32>, !transform.any_op
|
|
transform.debug.emit_param_as_remark %l1d_size, "L1d::size_in_bytes" at %func : !transform.param<i32>, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
#l1_size = #dlti.map<"size_in_bytes" = 65536 : i32>
|
|
#l1d_size = #dlti.map<"size_in_bytes" = 32768 : i32>
|
|
module attributes { test.dlti =
|
|
#dlti.target_system_spec<"CPU" =
|
|
#dlti.target_device_spec<"cache" =
|
|
#dlti.map<"L1" = #l1_size,
|
|
"L1d" = #l1d_size >>> } {
|
|
// expected-remark @below {{L1::size_in_bytes 65536 : i32}}
|
|
// expected-remark @below {{L1d::size_in_bytes 32768 : i32}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%l1_size = transform.dlti.query ["CPU","cache","L1","size_in_bytes"] at %func : (!transform.any_op) -> !transform.param<i32>
|
|
%l1d_size = transform.dlti.query ["CPU","cache","L1d","size_in_bytes"] at %func : (!transform.any_op) -> !transform.param<i32>
|
|
transform.debug.emit_param_as_remark %l1_size, "L1::size_in_bytes" at %func : !transform.param<i32>, !transform.any_op
|
|
transform.debug.emit_param_as_remark %l1d_size, "L1d::size_in_bytes" at %func : !transform.param<i32>, !transform.any_op
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<"inner_most_tile_size" = 42 : i32>> } {
|
|
// CHECK-LABEL: func @matmul_tensors
|
|
func.func @matmul_tensors(
|
|
%arg0: tensor<?x?xf32>, %arg1: tensor<?x?xf32>, %arg2: tensor<?x?xf32>)
|
|
-> tensor<?x?xf32> {
|
|
// CHECK: scf.for {{.*}} to {{.*}} step {{.*}}42
|
|
// CHECK: tensor.extract_slice
|
|
// CHECK: linalg.matmul
|
|
// CHECK: tensor.insert_slice
|
|
// CHECK: scf.yield
|
|
%0 = linalg.matmul ins(%arg0, %arg1: tensor<?x?xf32>, tensor<?x?xf32>)
|
|
outs(%arg2: tensor<?x?xf32>)
|
|
-> tensor<?x?xf32>
|
|
// CHECK: return
|
|
return %0 : tensor<?x?xf32>
|
|
}
|
|
}
|
|
|
|
// Demonstrates obtaining transform op parameters from DLTI attributes and directly putting them to use.
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%matmul = transform.structured.match ops{["linalg.matmul"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
%tile_size = transform.dlti.query ["CPU","inner_most_tile_size"] at %matmul : (!transform.any_op) -> !transform.param<i32>
|
|
transform.structured.tile_using_for %matmul tile_sizes [%tile_size] : (!transform.any_op, !transform.param<i32>) -> (!transform.any_op, !transform.any_op)
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{key "NPU" has no DLTI-mapping per attr: #dlti.target_system_spec}}
|
|
module attributes { test.dlti = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<"test.id" = 42 : i32>,
|
|
"GPU" = #dlti.target_device_spec<"test.id" = 43 : i32>> } {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["NPU","test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{key "unspecified" has no DLTI-mapping per attr: #dlti.target_device_spec}}
|
|
module attributes { test.dlti = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<"test.id" = 42 : i32>,
|
|
"GPU" = #dlti.target_device_spec<"test.id" = 43 : i32>> } {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["CPU","unspecified"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{key "test.id" has no DLTI-mapping per attr: #dlti.target_system_spec}}
|
|
module attributes { test.dlti = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<"test.id" = 42 : i32>,
|
|
"GPU" = #dlti.target_device_spec<"test.id" = 43 : i32>> } {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{key "CPU" has no DLTI-mapping per attr: #dlti.dl_spec}}
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32> } {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["CPU","test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{got non-DLTI-queryable attribute upon looking up keys ["CPU"]}}
|
|
module attributes { test.dlti = #dlti.dl_spec<"CPU" = 42 : i32> } {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["CPU","test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{got non-DLTI-queryable attribute upon looking up keys [i32]}}
|
|
module attributes { test.dlti = #dlti.dl_spec<i32 = 32 : i32> } {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query [i32,"width_in_bits"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
// expected-note @below {{no DLTI-queryable attrs on target op or any of its ancestors}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["CPU","test.id"] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-note @below {{key i64 has no DLTI-mapping per attr: #dlti.map<i32 = 32 : i64>}}
|
|
module attributes { test.dlti = #dlti.map<"width_in_bits" = #dlti.map<i32 = 32>>} {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query ["width_in_bits",i64] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.dl_spec<"test.id" = 42 : i32>} {
|
|
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
|
|
// expected-error @below {{'transform.dlti.query' keys of wrong type: only StringAttr and TypeAttr are allowed}}
|
|
%param = transform.dlti.query [1] at %funcs : (!transform.any_op) -> !transform.param<i64>
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.map<#dlti.dl_entry<"test.id", 42 : i32>>} {
|
|
// expected-error @below {{target op of failed DLTI query}}
|
|
// expected-note @below {{no keys provided to attempt query with}}
|
|
func.func private @f()
|
|
}
|
|
|
|
module attributes {transform.with_named_sequence} {
|
|
transform.named_sequence @__transform_main(%arg: !transform.any_op) {
|
|
%func = transform.structured.match ops{["func.func"]} in %arg : (!transform.any_op) -> !transform.any_op
|
|
// expected-error @below {{'transform.dlti.query' op failed to apply}}
|
|
%param = transform.dlti.query [] at %func : (!transform.any_op) -> !transform.any_param
|
|
transform.yield
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
module attributes { test.dlti = #dlti.dl_spec<#dlti.dl_entry<"test.id", 42 : i32>>} {
|
|
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
|
|
// expected-error @below {{expected the type of the parameter attribute ('i32') to match the parameter type ('i64')}}
|
|
%param = transform.dlti.query ["test.id"] at %funcs : (!transform.any_op) -> !transform.param<i64>
|
|
transform.yield
|
|
}
|
|
} |