d7c53a91c2
This commit pulls apart the inherent attribute dependence of classes like EnumAttrInfo and EnumAttrCase, factoring them out into simpler EnumCase and EnumInfo variants. This allows specifying the cases of an enum without needing to make the cases, or the EnumInfo itself, a subclass of SignlessIntegerAttrBase. The existing classes are retained as subclasses of the new ones, both for backwards compatibility and to allow attribute-specific information. In addition, the new BitEnum class changes its default printer/parser behavior: cases when multiple keywords appear, like having both nuw and nsw in overflow flags, will no longer be quoted by the operator<<, and the FieldParser instance will now expect multiple keywords. All instances of BitEnumAttr retain the old behavior.
953 lines
32 KiB
MLIR
953 lines
32 KiB
MLIR
// RUN: mlir-opt %s -split-input-file -allow-unregistered-dialect -verify-diagnostics | FileCheck %s
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test AnyAttrOf attributes
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @any_attr_of_pass() {
|
|
"test.any_attr_of_i32_str"() {
|
|
// CHECK: attr = 3 : i32
|
|
attr = 3 : i32
|
|
} : () -> ()
|
|
|
|
"test.any_attr_of_i32_str"() {
|
|
// CHECK: attr = "string_data"
|
|
attr = "string_data"
|
|
} : () -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @any_attr_of_fail() {
|
|
// expected-error @+1 {{'test.any_attr_of_i32_str' op attribute 'attr' failed to satisfy constraint: 32-bit signless integer attribute or string attribute}}
|
|
"test.any_attr_of_i32_str"() {
|
|
attr = 3 : i64
|
|
} : () -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test float attributes
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @float_attrs_pass() {
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f4E2M1FN
|
|
float_attr = 2. : f4E2M1FN
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f6E2M3FN
|
|
float_attr = 2. : f6E2M3FN
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f6E3M2FN
|
|
float_attr = 2. : f6E3M2FN
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E5M2
|
|
float_attr = 2. : f8E5M2
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E4M3
|
|
float_attr = 2. : f8E4M3
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E4M3FN
|
|
float_attr = 2. : f8E4M3FN
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E5M2FNUZ
|
|
float_attr = 2. : f8E5M2FNUZ
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E4M3FNUZ
|
|
float_attr = 2. : f8E4M3FNUZ
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E4M3B11FNUZ
|
|
float_attr = 2. : f8E4M3B11FNUZ
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E3M4
|
|
float_attr = 2. : f8E3M4
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f8E8M0FNU
|
|
float_attr = 2. : f8E8M0FNU
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f16
|
|
float_attr = 2. : f16
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : bf16
|
|
float_attr = 2. : bf16
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : tf32
|
|
float_attr = 2. : tf32
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f32
|
|
float_attr = 2. : f32
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f64
|
|
float_attr = 2. : f64
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f80
|
|
float_attr = 2. : f80
|
|
} : () -> ()
|
|
"test.float_attrs"() {
|
|
// CHECK: float_attr = 2.000000e+00 : f128
|
|
float_attr = 2. : f128
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test integer attributes
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @int_attrs_pass() {
|
|
"test.int_attrs"() {
|
|
// CHECK: any_i32_attr = 5 : ui32
|
|
any_i32_attr = 5 : ui32,
|
|
// CHECK-SAME: index_attr = 8 : index
|
|
index_attr = 8 : index,
|
|
// CHECK-SAME: si32_attr = 7 : si32
|
|
si32_attr = 7 : si32,
|
|
// CHECK-SAME: ui32_attr = 6 : ui32
|
|
ui32_attr = 6 : ui32
|
|
} : () -> ()
|
|
|
|
"test.int_attrs"() {
|
|
// CHECK: any_i32_attr = 5 : si32
|
|
any_i32_attr = 5 : si32,
|
|
index_attr = 8 : index,
|
|
si32_attr = 7 : si32,
|
|
ui32_attr = 6 : ui32
|
|
} : () -> ()
|
|
|
|
"test.int_attrs"() {
|
|
// CHECK: any_i32_attr = 5 : i32
|
|
any_i32_attr = 5 : i32,
|
|
index_attr = 8 : index,
|
|
si32_attr = 7 : si32,
|
|
ui32_attr = 6 : ui32
|
|
} : () -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Check that the maximum and minimum integer attribute values are
|
|
// representable and preserved during a round-trip.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @int_attrs_pass() {
|
|
"test.in_range_attrs"() {
|
|
// CHECK: attr_00 = -128 : i8
|
|
attr_00 = -128 : i8,
|
|
// CHECK-SAME: attr_01 = 127 : i8
|
|
attr_01 = 127 : i8,
|
|
// CHECK-SAME: attr_02 = -128 : si8
|
|
attr_02 = -128 : si8,
|
|
// CHECK-SAME: attr_03 = 127 : si8
|
|
attr_03 = 127 : si8,
|
|
// CHECK-SAME: attr_04 = 255 : ui8
|
|
attr_04 = 255 : ui8,
|
|
// CHECK-SAME: attr_05 = -32768 : i16
|
|
attr_05 = -32768 : i16,
|
|
// CHECK-SAME: attr_06 = 32767 : i16
|
|
attr_06 = 32767 : i16,
|
|
// CHECK-SAME: attr_07 = -32768 : si16
|
|
attr_07 = -32768 : si16,
|
|
// CHECK-SAME: attr_08 = 32767 : si16
|
|
attr_08 = 32767 : si16,
|
|
// CHECK-SAME: attr_09 = 65535 : ui16
|
|
attr_09 = 65535 : ui16,
|
|
// CHECK-SAME: attr_10 = -2147483647 : i32
|
|
attr_10 = -2147483647 : i32,
|
|
// CHECK-SAME: attr_11 = 2147483646 : i32
|
|
attr_11 = 2147483646 : i32,
|
|
// CHECK-SAME: attr_12 = -2147483647 : si32
|
|
attr_12 = -2147483647 : si32,
|
|
// CHECK-SAME: attr_13 = 2147483646 : si32
|
|
attr_13 = 2147483646 : si32,
|
|
// CHECK-SAME: attr_14 = 4294967295 : ui32
|
|
attr_14 = 4294967295 : ui32,
|
|
// CHECK-SAME: attr_15 = -9223372036854775808 : i64
|
|
attr_15 = -9223372036854775808 : i64,
|
|
// CHECK-SAME: attr_16 = 9223372036854775807 : i64
|
|
attr_16 = 9223372036854775807 : i64,
|
|
// CHECK-SAME: attr_17 = -9223372036854775808 : si64
|
|
attr_17 = -9223372036854775808 : si64,
|
|
// CHECK-SAME: attr_18 = 9223372036854775807 : si64
|
|
attr_18 = 9223372036854775807 : si64,
|
|
// CHECK-SAME: attr_19 = 18446744073709551615 : ui64
|
|
attr_19 = 18446744073709551615 : ui64,
|
|
// CHECK-SAME: attr_20 = 1 : ui1
|
|
attr_20 = 1 : ui1,
|
|
// CHECK-SAME: attr_21 = -1 : si1
|
|
attr_21 = -1 : si1,
|
|
// CHECK-SAME: attr_22 = 79228162514264337593543950335 : ui96
|
|
attr_22 = 79228162514264337593543950335 : ui96,
|
|
// CHECK-SAME: attr_23 = -39614081257132168796771975168 : si96
|
|
attr_23 = -39614081257132168796771975168 : si96
|
|
} : () -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Check that positive values larger than 2^n-1 for signless integers
|
|
// are mapped to their negative signed counterpart. This behaviour is
|
|
// undocumented in the language specification, but it is what the
|
|
// parser currently does.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @int_attrs_pass() {
|
|
"test.i8_attr"() {
|
|
// CHECK: attr_00 = -1 : i8
|
|
attr_00 = 255 : i8,
|
|
// CHECK-SAME: attr_01 = -1 : i16
|
|
attr_01 = 65535 : i16,
|
|
// CHECK-SAME: attr_02 = -1 : i32
|
|
attr_02 = 4294967295 : i32,
|
|
// CHECK-SAME: attr_03 = -1 : i64
|
|
attr_03 = 18446744073709551615 : i64
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Check that i0 is parsed and verified correctly. It can only have value 0.
|
|
// We check it explicitly because there are various special cases for it that
|
|
// are good to verify.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @int0_attrs_pass() {
|
|
"test.i0_attr"() {
|
|
// CHECK: attr_00 = 0 : i0
|
|
attr_00 = 0 : i0,
|
|
// CHECK: attr_01 = 0 : si0
|
|
attr_01 = 0 : si0,
|
|
// CHECK: attr_02 = 0 : ui0
|
|
attr_02 = 0 : ui0,
|
|
// CHECK: attr_03 = 0 : i0
|
|
attr_03 = 0x0000 : i0,
|
|
// CHECK: attr_04 = 0 : si0
|
|
attr_04 = 0x0000 : si0,
|
|
// CHECK: attr_05 = 0 : ui0
|
|
attr_05 = 0x0000 : ui0
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @int0_attrs_negative_fail() {
|
|
"test.i0_attr"() {
|
|
// expected-error @+1 {{integer constant out of range for attribute}}
|
|
attr_00 = -1 : i0
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @int0_attrs_positive_fail() {
|
|
"test.i0_attr"() {
|
|
// expected-error @+1 {{integer constant out of range for attribute}}
|
|
attr_00 = 1 : i0
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_int_attrs_signedness_fail() {
|
|
// expected-error @+1 {{'si32_attr' failed to satisfy constraint: 32-bit signed integer attribute}}
|
|
"test.int_attrs"() {
|
|
any_i32_attr = 5 : i32,
|
|
index_attr = 8 : index,
|
|
si32_attr = 7 : ui32,
|
|
ui32_attr = 6 : ui32
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_int_attrs_signedness_fail() {
|
|
// expected-error @+1 {{'ui32_attr' failed to satisfy constraint: 32-bit unsigned integer attribute}}
|
|
"test.int_attrs"() {
|
|
any_i32_attr = 5 : i32,
|
|
index_attr = 8 : index,
|
|
si32_attr = 7 : si32,
|
|
ui32_attr = 6 : si32
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_int_attrs_type_fail() {
|
|
// expected-error @+1 {{'any_i32_attr' failed to satisfy constraint: 32-bit integer attribute}}
|
|
"test.int_attrs"() {
|
|
any_i32_attr = 5.0 : f32,
|
|
si32_attr = 7 : si32,
|
|
ui32_attr = 6 : ui32,
|
|
index_attr = 1 : index
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test Non-negative Int Attr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @non_negative_int_attr_pass() {
|
|
// CHECK: test.non_negative_int_attr
|
|
"test.non_negative_int_attr"() {i32attr = 5 : i32, i64attr = 10 : i64} : () -> ()
|
|
// CHECK: test.non_negative_int_attr
|
|
"test.non_negative_int_attr"() {i32attr = 0 : i32, i64attr = 0 : i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @negative_int_attr_fail() {
|
|
// expected-error @+1 {{'i32attr' failed to satisfy constraint: 32-bit signless integer attribute whose value is non-negative}}
|
|
"test.non_negative_int_attr"() {i32attr = -5 : i32, i64attr = 10 : i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @negative_int_attr_fail() {
|
|
// expected-error @+1 {{'i64attr' failed to satisfy constraint: 64-bit signless integer attribute whose value is non-negative}}
|
|
"test.non_negative_int_attr"() {i32attr = 5 : i32, i64attr = -10 : i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test Positive Int Attr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @positive_int_attr_pass() {
|
|
// CHECK: test.positive_int_attr
|
|
"test.positive_int_attr"() {i32attr = 5 : i32, i64attr = 10 : i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @positive_int_attr_fail() {
|
|
// expected-error @+1 {{'i32attr' failed to satisfy constraint: 32-bit signless integer attribute whose value is positive}}
|
|
"test.positive_int_attr"() {i32attr = 0 : i32, i64attr = 5: i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @positive_int_attr_fail() {
|
|
// expected-error @+1 {{'i64attr' failed to satisfy constraint: 64-bit signless integer attribute whose value is positive}}
|
|
"test.positive_int_attr"() {i32attr = 5 : i32, i64attr = 0: i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @positive_int_attr_fail() {
|
|
// expected-error @+1 {{'i32attr' failed to satisfy constraint: 32-bit signless integer attribute whose value is positive}}
|
|
"test.positive_int_attr"() {i32attr = -10 : i32, i64attr = 5 : i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @positive_int_attr_fail() {
|
|
// expected-error @+1 {{'i64attr' failed to satisfy constraint: 64-bit signless integer attribute whose value is positive}}
|
|
"test.positive_int_attr"() {i32attr = 5 : i32, i64attr = -10 : i64} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test TypeArrayAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @correct_type_array_attr_pass() {
|
|
// CHECK: test.type_array_attr
|
|
"test.type_array_attr"() {attr = [i32, f32]} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @non_type_in_type_array_attr_fail() {
|
|
// expected-error @+1 {{'attr' failed to satisfy constraint: type array attribute}}
|
|
"test.type_array_attr"() {attr = [i32, 5 : i64]} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test StringAttr with custom type
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// CHECK-LABEL: func @string_attr_custom_type_valid
|
|
func.func @string_attr_custom_type_valid() {
|
|
// CHECK: "string_data" : i64
|
|
test.string_attr_with_type "string_data" : i64
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @string_attr_custom_type_invalid() {
|
|
// expected-error @+1 {{'attr' failed to satisfy constraint: string attribute of integer}}
|
|
test.string_attr_with_type "string_data" : f32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @string_attr_custom_mixed_type
|
|
func.func @string_attr_custom_mixed_type() {
|
|
// CHECK: "string_data" : i64
|
|
test.string_attr_with_mixed_type "string_data" : i64
|
|
// CHECK: 42 : i64
|
|
test.string_attr_with_mixed_type 42 : i64
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test I32EnumAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// CHECK-LABEL: func @allowed_cases_pass
|
|
func.func @allowed_cases_pass() {
|
|
// CHECK: test.i32_enum_attr
|
|
%0 = "test.i32_enum_attr"() {attr = 5: i32} : () -> i32
|
|
// CHECK: test.i32_enum_attr
|
|
%1 = "test.i32_enum_attr"() {attr = 10: i32} : () -> i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @disallowed_case7_fail() {
|
|
// expected-error @+1 {{allowed 32-bit signless integer cases: 5, 10}}
|
|
%0 = "test.i32_enum_attr"() {attr = 7: i32} : () -> i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @disallowed_case7_fail() {
|
|
// expected-error @+1 {{allowed 32-bit signless integer cases: 5, 10}}
|
|
%0 = "test.i32_enum_attr"() {attr = 5: i64} : () -> i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test I64EnumAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// CHECK-LABEL: func @allowed_cases_pass
|
|
func.func @allowed_cases_pass() {
|
|
// CHECK: test.i64_enum_attr
|
|
%0 = "test.i64_enum_attr"() {attr = 5: i64} : () -> i32
|
|
// CHECK: test.i64_enum_attr
|
|
%1 = "test.i64_enum_attr"() {attr = 10: i64} : () -> i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @disallowed_case7_fail() {
|
|
// expected-error @+1 {{allowed 64-bit signless integer cases: 5, 10}}
|
|
%0 = "test.i64_enum_attr"() {attr = 7: i64} : () -> i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @disallowed_case7_fail() {
|
|
// expected-error @+1 {{allowed 64-bit signless integer cases: 5, 10}}
|
|
%0 = "test.i64_enum_attr"() {attr = 5: i32} : () -> i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test BitEnumAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// CHECK-LABEL: func @allowed_cases_pass
|
|
func.func @allowed_cases_pass() {
|
|
// CHECK: test.op_with_bit_enum <read, write>
|
|
"test.op_with_bit_enum"() {value = #test.bit_enum<read, write>} : () -> ()
|
|
// CHECK: test.op_with_bit_enum <read, execute>
|
|
test.op_with_bit_enum <read,execute>
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @allowed_cases_pass
|
|
func.func @allowed_cases_pass() {
|
|
// CHECK: test.op_with_bit_enum_vbar <user | group>
|
|
"test.op_with_bit_enum_vbar"() {
|
|
value = #test.bit_enum_vbar<user|group>
|
|
} : () -> ()
|
|
// CHECK: test.op_with_bit_enum_vbar <user | group | other>
|
|
test.op_with_bit_enum_vbar <user | group | other>
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @disallowed_case_sticky_fail() {
|
|
// expected-error@+2 {{expected one of [read, write, execute] for a test bit enum, got: sticky}}
|
|
// expected-error@+1 {{failed to parse TestBitEnumAttr}}
|
|
"test.op_with_bit_enum"() {value = #test.bit_enum<sticky>} : () -> ()
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test FloatElementsAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @correct_type_pass() {
|
|
"test.float_elements_attr"() {
|
|
// CHECK: scalar_f32_attr = dense<5.000000e+00> : tensor<2xf32>
|
|
// CHECK: tensor_f64_attr = dense<6.000000e+00> : tensor<4x8xf64>
|
|
scalar_f32_attr = dense<5.0> : tensor<2xf32>,
|
|
tensor_f64_attr = dense<6.0> : tensor<4x8xf64>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_element_type_pass() {
|
|
// expected-error @+1 {{failed to satisfy constraint: 32-bit float elements attribute of shape [2]}}
|
|
"test.float_elements_attr"() {
|
|
scalar_f32_attr = dense<5.0> : tensor<2xf64>,
|
|
tensor_f64_attr = dense<6.0> : tensor<4x8xf64>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @correct_type_pass() {
|
|
// expected-error @+1 {{failed to satisfy constraint: 64-bit float elements attribute of shape [4, 8]}}
|
|
"test.float_elements_attr"() {
|
|
scalar_f32_attr = dense<5.0> : tensor<2xf32>,
|
|
tensor_f64_attr = dense<6.0> : tensor<4xf64>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @tf32_elements_attr() {
|
|
// CHECK: "foo"() {attr = dense<4.000000e+00> : tensor<tf32>} : () -> ()
|
|
"foo"() {attr = dense<4.0> : tensor<tf32>} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test StringElementsAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @simple_scalar_example() {
|
|
"test.string_elements_attr"() {
|
|
// CHECK: dense<"example">
|
|
scalar_string_attr = dense<"example"> : tensor<2x!unknown<"">>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @escape_string_example() {
|
|
"test.string_elements_attr"() {
|
|
// CHECK: dense<"new\0Aline">
|
|
scalar_string_attr = dense<"new\nline"> : tensor<2x!unknown<"">>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple_scalar_example() {
|
|
"test.string_elements_attr"() {
|
|
// CHECK: dense<["example1", "example2"]>
|
|
scalar_string_attr = dense<["example1", "example2"]> : tensor<2x!unknown<"">>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test DenseArrayAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// CHECK-LABEL: func @dense_array_attr
|
|
func.func @dense_array_attr() attributes {
|
|
// CHECK-SAME: emptyf32attr = array<f32>,
|
|
emptyf32attr = array<f32>,
|
|
// CHECK-SAME: emptyf64attr = array<f64>,
|
|
emptyf64attr = array<f64>,
|
|
// CHECK-SAME: emptyi16attr = array<i16>,
|
|
emptyi16attr = array<i16>,
|
|
// CHECK-SAME: emptyi1attr = array<i1>,
|
|
emptyi1attr = array<i1>,
|
|
// CHECK-SAME: emptyi32attr = array<i32>,
|
|
emptyi32attr = array<i32>,
|
|
// CHECK-SAME: emptyi64attr = array<i64>,
|
|
emptyi64attr = array<i64>,
|
|
// CHECK-SAME: emptyi8attr = array<i8>,
|
|
emptyi8attr = array<i8>,
|
|
// CHECK-SAME: f32attr = array<f32: 1.024000e+03, 4.530000e+02, -6.435000e+03>,
|
|
f32attr = array<f32: 1024., 453., -6435.>,
|
|
// CHECK-SAME: f64attr = array<f64: -1.420000e+02>,
|
|
f64attr = array<f64: -142.>,
|
|
// CHECK-SAME: i16attr = array<i16: 3, 5, -4, 10>,
|
|
i16attr = array<i16: 3, 5, -4, 10>,
|
|
// CHECK-SAME: i1attr = array<i1: true, false, true>,
|
|
i1attr = array<i1: true, false, true>,
|
|
// CHECK-SAME: i32attr = array<i32: 1024, 453, -6435>,
|
|
i32attr = array<i32: 1024, 453, -6435>,
|
|
// CHECK-SAME: i64attr = array<i64: -142>,
|
|
i64attr = array<i64: -142>,
|
|
// CHECK-SAME: i8attr = array<i8: 1, -2, 3>
|
|
i8attr = array<i8: 1, -2, 3>
|
|
} {
|
|
// CHECK: test.dense_array_attr
|
|
test.dense_array_attr
|
|
// CHECK-SAME: i1attr = [true, false, true]
|
|
i1attr = [true, false, true]
|
|
// CHECK-SAME: i8attr = [1, -2, 3]
|
|
i8attr = [1, -2, 3]
|
|
// CHECK-SAME: i16attr = [3, 5, -4, 10]
|
|
i16attr = [3, 5, -4, 10]
|
|
// CHECK-SAME: i32attr = [1024, 453, -6435]
|
|
i32attr = [1024, 453, -6435]
|
|
// CHECK-SAME: i64attr = [-142]
|
|
i64attr = [-142]
|
|
// CHECK-SAME: f32attr = [1.024000e+03, 4.530000e+02, -6.435000e+03]
|
|
f32attr = [1024., 453., -6435.]
|
|
// CHECK-SAME: f64attr = [-1.420000e+02]
|
|
f64attr = [-142.]
|
|
// CHECK-SAME: emptyattr = []
|
|
emptyattr = []
|
|
|
|
// CHECK: array.sizes
|
|
// CHECK-SAME: i0 = array<i0: 0, 0>
|
|
// CHECK-SAME: ui0 = array<ui0: 0, 0>
|
|
// CHECK-SAME: si0 = array<si0: 0, 0>
|
|
// CHECK-SAME: i24 = array<i24: -42, 42, 8388607>
|
|
// CHECK-SAME: ui24 = array<ui24: 16777215>
|
|
// CHECK-SAME: si24 = array<si24: -8388608>
|
|
// CHECK-SAME: bf16 = array<bf16: 1.2{{[0-9]+}}e+00, 3.4{{[0-9]+}}e+00>
|
|
// CHECK-SAME: f16 = array<f16: 1.{{[0-9]+}}e+00, 3.{{[0-9]+}}e+00>
|
|
"array.sizes"() {
|
|
x0_i0 = array<i0: 0, 0>,
|
|
x1_ui0 = array<ui0: 0, 0>,
|
|
x2_si0 = array<si0: 0, 0>,
|
|
x3_i24 = array<i24: -42, 42, 8388607>,
|
|
x4_ui24 = array<ui24: 16777215>,
|
|
x5_si24 = array<si24: -8388608>,
|
|
x6_bf16 = array<bf16: 1.2, 3.4>,
|
|
x7_f16 = array<f16: 1., 3.>
|
|
}: () -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @test_invalid_bitwidth_type() {
|
|
// expected-error @below{{element type bitwidth must be a multiple of 8}}
|
|
"foo"() {tf32attr = array<tf32: 1024.0>} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedDenseArrayAttr() {
|
|
"test.confined_dense_array_attr"() {
|
|
i64attr = array<i64: 0, 2, 3>,
|
|
i32attr = array<i32: 1>,
|
|
emptyattr = array<i16>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedDenseArrayAttrDuplicateValues() {
|
|
// expected-error@+1{{'test.confined_dense_array_attr' op attribute 'i64attr' failed to satisfy constraint: i64 dense array attribute should be in increasing order}}
|
|
"test.confined_dense_array_attr"() {
|
|
emptyattr = array<i16>,
|
|
i32attr = array<i32: 1, 1>,
|
|
i64attr = array<i64: 0, 2, 2>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedDenseArrayAttrDecreasingOrder() {
|
|
// expected-error@+1{{'test.confined_dense_array_attr' op attribute 'i32attr' failed to satisfy constraint: i32 dense array attribute should be in non-decreasing order}}
|
|
"test.confined_dense_array_attr"() {
|
|
emptyattr = array<i16>,
|
|
i32attr = array<i32: 1, 0>,
|
|
i64attr = array<i64: 0, 2, 3>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedStrictlyPositiveDenseArrayAttr() {
|
|
"test.confined_strictly_positive_attr"() {
|
|
i8attr = array<i8: 2, 3>,
|
|
i16attr = array<i16: 20, 30>,
|
|
i32attr = array<i32: 1>,
|
|
i64attr = array<i64: 1, 2, 3>,
|
|
f32attr = array<f32: 1.1, 2.1>,
|
|
f64attr = array<f64: 2.1, 3.1>,
|
|
emptyattr = array<i16>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedStrictlyPositiveDenseArrayAttr() {
|
|
// expected-error@+1{{'test.confined_strictly_positive_attr' op attribute 'i64attr' failed to satisfy constraint: i64 dense array attribute whose value is positive}}
|
|
"test.confined_strictly_positive_attr"() {
|
|
i8attr = array<i8: 2, 3>,
|
|
i16attr = array<i16: 20, 30>,
|
|
i32attr = array<i32: 1>,
|
|
i64attr = array<i64: 0, 2, 3>,
|
|
f32attr = array<f32: 1.1, 2.1>,
|
|
f64attr = array<f64: 2.1, 3.1>,
|
|
emptyattr = array<i16>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedNonNegativeDenseArrayAttr() {
|
|
"test.confined_non_negative_attr"() {
|
|
i8attr = array<i8: 0, 3>,
|
|
i16attr = array<i16: 0, 30>,
|
|
i32attr = array<i32: 1>,
|
|
i64attr = array<i64: 1, 0, 3>,
|
|
f32attr = array<f32: 0.0, 2.1>,
|
|
f64attr = array<f64: 0.0, 3.1>,
|
|
emptyattr = array<i16>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @testConfinedNonNegativeDenseArrayAttr() {
|
|
// expected-error@+1{{'test.confined_non_negative_attr' op attribute 'i64attr' failed to satisfy constraint: i64 dense array attribute whose value is non-negative}}
|
|
"test.confined_non_negative_attr"() {
|
|
i8attr = array<i8: 0, 3>,
|
|
i16attr = array<i16: 0, 30>,
|
|
i32attr = array<i32: 1>,
|
|
i64attr = array<i64: -1, 0, 3>,
|
|
f32attr = array<f32: 0.0, 2.1>,
|
|
f64attr = array<f64: 0.0, 3.1>,
|
|
emptyattr = array<i16>
|
|
} : () -> ()
|
|
func.return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test SymbolRefAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @fn() { return }
|
|
|
|
// CHECK: test.symbol_ref_attr
|
|
"test.symbol_ref_attr"() {symbol = @fn} : () -> ()
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test IntElementsAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @correct_int_elements_attr_pass() {
|
|
"test.int_elements_attr"() {
|
|
// CHECK: any_i32_attr = dense<5> : tensor<1x2x3x4xui32>,
|
|
any_i32_attr = dense<5> : tensor<1x2x3x4xui32>,
|
|
i32_attr = dense<5> : tensor<6xi32>
|
|
} : () -> ()
|
|
|
|
"test.int_elements_attr"() {
|
|
// CHECK: any_i32_attr = dense<5> : tensor<1x2x3x4xsi32>,
|
|
any_i32_attr = dense<5> : tensor<1x2x3x4xsi32>,
|
|
i32_attr = dense<5> : tensor<6xi32>
|
|
} : () -> ()
|
|
|
|
"test.int_elements_attr"() {
|
|
// CHECK: any_i32_attr = dense<5> : tensor<1x2x3x4xi32>,
|
|
any_i32_attr = dense<5> : tensor<1x2x3x4xi32>,
|
|
i32_attr = dense<5> : tensor<6xi32>
|
|
} : () -> ()
|
|
|
|
"test.index_elements_attr"() {
|
|
// CHECK: any_index_attr = dense<5> : tensor<1x2x3x4xindex>,
|
|
any_index_attr = dense<5> : tensor<1x2x3x4xindex>,
|
|
index_attr = dense<5> : tensor<6xindex>
|
|
} : () -> ()
|
|
|
|
"test.hex_index_elements_attr"() {
|
|
// CHECK: hex_index_attr = dense<"0x00000C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000080C0000000000001A150000000000005078000000000000B659010000000000ECBC010000000000FEC5010000000000342902000000000046320200000000007C950200000000008E9E020000000000C401030000000000D60A0300000000000C6E0300000000001E7703000000000054DA03000000000066E30300000000009C46040000000000AE4F040000000000E4B2040000000000F6BB0400000000002C1F050000000000628100000000000098E40000000000000E0C00000000000020150000000000005678000000000000BC59010000000000F2BC01000000000004C60100000000003A290200000000004C320200000000008295020000000000949E020000000000CA01030000000000DC0A030000000000126E03000000000024770300000000005ADA0300000000006CE3030000000000A246040000000000B44F040000000000EAB2040000000000FCBB040000000000321F05000000000068810000000000009EE40000000000"> : tensor<23x5xindex>
|
|
hex_index_attr = dense<"0x00000C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000080C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000100C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000180C000000000000080C0000000000001A150000000000005078000000000000B659010000000000ECBC010000000000FEC5010000000000342902000000000046320200000000007C950200000000008E9E020000000000C401030000000000D60A0300000000000C6E0300000000001E7703000000000054DA03000000000066E30300000000009C46040000000000AE4F040000000000E4B2040000000000F6BB0400000000002C1F050000000000628100000000000098E40000000000000E0C00000000000020150000000000005678000000000000BC59010000000000F2BC01000000000004C60100000000003A290200000000004C320200000000008295020000000000949E020000000000CA01030000000000DC0A030000000000126E03000000000024770300000000005ADA0300000000006CE3030000000000A246040000000000B44F040000000000EAB2040000000000FCBB040000000000321F05000000000068810000000000009EE40000000000"> : tensor<23x5xindex>
|
|
} : () -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_int_elements_attr_type_fail() {
|
|
// expected-error @+1 {{'any_i32_attr' failed to satisfy constraint: 32-bit integer elements attribute}}
|
|
"test.int_elements_attr"() {
|
|
any_i32_attr = dense<5.0> : tensor<1x2x3x4xf32>,
|
|
i32_attr = dense<5> : tensor<6xi32>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_int_elements_attr_signedness_fail() {
|
|
// expected-error @+1 {{'i32_attr' failed to satisfy constraint: 32-bit signless integer elements attribute}}
|
|
"test.int_elements_attr"() {
|
|
any_i32_attr = dense<5> : tensor<1x2x3x4xi32>,
|
|
i32_attr = dense<5> : tensor<6xsi32>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test Ranked IntElementsAttr
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func.func @correct_type_pass() {
|
|
"test.ranked_int_elements_attr"() {
|
|
// CHECK: matrix_i64_attr = dense<6> : tensor<4x8xi64>
|
|
// CHECK: vector_i32_attr = dense<5> : tensor<2xi32>
|
|
matrix_i64_attr = dense<6> : tensor<4x8xi64>,
|
|
vector_i32_attr = dense<5> : tensor<2xi32>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_element_type_fail() {
|
|
// expected-error @+1 {{failed to satisfy constraint: 32-bit signless int elements attribute of shape [2]}}
|
|
"test.ranked_int_elements_attr"() {
|
|
matrix_i64_attr = dense<6> : tensor<4x8xi64>,
|
|
vector_i32_attr = dense<5> : tensor<2xi64>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_shape_fail() {
|
|
// expected-error @+1 {{failed to satisfy constraint: 64-bit signless int elements attribute of shape [4, 8]}}
|
|
"test.ranked_int_elements_attr"() {
|
|
matrix_i64_attr = dense<6> : tensor<4xi64>,
|
|
vector_i32_attr = dense<5> : tensor<2xi32>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @wrong_shape_fail() {
|
|
// expected-error @+1 {{failed to satisfy constraint: 32-bit signless int elements attribute of shape [2]}}
|
|
"test.ranked_int_elements_attr"() {
|
|
matrix_i64_attr = dense<6> : tensor<4x8xi64>,
|
|
vector_i32_attr = dense<5> : tensor<i32>
|
|
} : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Test DefaultValuedAttr Printing
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// CHECK-LABEL: @default_value_printing
|
|
func.func @default_value_printing(%arg0 : i32) {
|
|
// The attribute SHOULD NOT be printed because it is equal to the default
|
|
// CHECK: test.default_value_print %arg0
|
|
"test.default_value_print"(%arg0) {"value_with_default" = 0 : i32} : (i32) -> ()
|
|
// The attribute SHOULD be printed because it is not equal to the default
|
|
// CHECK: test.default_value_print {value_with_default = 1 : i32} %arg0
|
|
"test.default_value_print"(%arg0) {"value_with_default" = 1 : i32} : (i32) -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @type_attr_of_fail() {
|
|
// expected-error @below {{failed to satisfy constraint: type attribute of 64-bit signless integer}}
|
|
test.type_attr_of i32
|
|
return
|
|
}
|