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.
255 lines
8.6 KiB
MLIR
255 lines
8.6 KiB
MLIR
// RUN: mlir-opt -split-input-file %s | FileCheck %s
|
|
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.map = #dlti.map<
|
|
// CHECK-SAME: "magic_num" = 42 : i32,
|
|
// CHECK-SAME: "magic_num_float" = 4.242000e+01 : f32,
|
|
// CHECK-SAME: "magic_type" = i32,
|
|
// CHECK-SAME: i32 = #dlti.map<"bitwidth" = 32 : i32>
|
|
// CHECK: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.map = #dlti.map<"magic_num" = 42 : i32,
|
|
"magic_num_float" = 42.42 : f32,
|
|
"magic_type" = i32,
|
|
i32 = #dlti.map<"bitwidth" = 32 : i32>>
|
|
} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.map = #dlti.map<
|
|
// CHECK-SAME: "magic_num" = 42 : i32,
|
|
// CHECK-SAME: "magic_num_float" = 4.242000e+01 : f32,
|
|
// CHECK-SAME: "magic_type" = i32,
|
|
// CHECK-SAME: i32 = #dlti.map<"bitwidth" = 32 : i32>
|
|
// CHECK: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.map = #dlti.map<
|
|
#dlti.dl_entry<"magic_num", 42 : i32>,
|
|
#dlti.dl_entry<"magic_num_float", 42.42 : f32>,
|
|
#dlti.dl_entry<"magic_type", i32>,
|
|
#dlti.dl_entry<i32, #dlti.map<#dlti.dl_entry<"bitwidth", 32 : i32>>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.map = #dlti.map<
|
|
// CHECK-SAME: "CPU" = #dlti.map<"L1_cache_size_in_bytes" = 4096 : i32>,
|
|
// CHECK-SAME: "GPU" = #dlti.map<"max_vector_op_width" = 128 : i32>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.map = #dlti.map<
|
|
#dlti.dl_entry<"CPU", #dlti.map<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 4096 : i32>>>,
|
|
#dlti.dl_entry<"GPU", #dlti.map<
|
|
#dlti.dl_entry<"max_vector_op_width", 128 : i32>>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 4096 : i32>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 128 : i32>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 4096 : i32>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 128 : i32>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 4096 : i32>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 8192 : i32>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 4096 : i32>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 8192 : i32>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 4096 : i64>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 8192 : i64>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 4096 : i64>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 8192 : i64>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 64 : i32>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 128 : i32>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 64 : i32>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 128 : i32>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 64 : i64>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 128 : i64>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 64 : i64>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 128 : i64>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 64 : i64>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 128 : i64>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 64 : i64>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 128 : i64>>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// Check values of mixed type
|
|
//
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 4096 : ui32>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = "128">
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", 4096 : ui32>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", "128">>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// Check values of mixed type
|
|
//
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 4.096000e+03 : f32>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = "128">
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"max_vector_op_width", 4096.0 : f32>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", "128">>
|
|
>} {}
|
|
|
|
|
|
// -----
|
|
|
|
// Check values of mixed type
|
|
//
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "vector_unit" = #dlti.map<
|
|
// CHECK-SAME: "max_op_width" = 4.096000e+03 : f32>>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = "128">
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"vector_unit", #dlti.map<
|
|
#dlti.dl_entry<"max_op_width", 4096.0 : f32>>>>,
|
|
"GPU" = #dlti.target_device_spec<
|
|
#dlti.dl_entry<"L1_cache_size_in_bytes", "128">>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// Check values of mixed type
|
|
//
|
|
// CHECK: module attributes {
|
|
// CHECK-SAME: dlti.target_system_spec = #dlti.target_system_spec<
|
|
// CHECK-SAME: "CPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "L1_cache_size_in_bytes" = 4096 : ui32>,
|
|
// CHECK-SAME: "GPU" = #dlti.target_device_spec<
|
|
// CHECK-SAME: "max_vector_op_width" = 128 : i64>
|
|
// CHECK-SAME: >} {
|
|
// CHECK: }
|
|
module attributes {
|
|
dlti.target_system_spec = #dlti.target_system_spec<
|
|
"CPU" = #dlti.target_device_spec<"L1_cache_size_in_bytes" = 4096 : ui32>,
|
|
"GPU" = #dlti.target_device_spec<"max_vector_op_width" = 128>
|
|
>} {}
|
|
|
|
// -----
|
|
|
|
// CHECK: "test.op_with_dlti_map"() ({
|
|
// CHECK: }) {dlti.map = #dlti.map<"dlti.unknown_id" = 42 : i64>}
|
|
"test.op_with_dlti_map"() ({
|
|
}) { dlti.map = #dlti.map<#dlti.dl_entry<"dlti.unknown_id", 42>> } : () -> ()
|
|
|
|
// -----
|
|
|
|
// CHECK: "test.op_with_dlti_map"() ({
|
|
// CHECK: }) {dlti.map = #dlti.map<i32 = 42 : i64>}
|
|
"test.op_with_dlti_map"() ({
|
|
}) { dlti.map = #dlti.map<#dlti.dl_entry<i32, 42>> } : () -> ()
|