Skip to content

Commit 5c77e4d

Browse files
committed
adding equality tests
1 parent 9d509b3 commit 5c77e4d

File tree

1 file changed

+34
-1
lines changed

1 file changed

+34
-1
lines changed

mlir/test/Integration/Dialect/Arith/CPU/comparison.mlir

Lines changed: 34 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ func.func @cmpi_sge_i1(%v1 : i1, %v2 : i1) {
3939
return
4040
}
4141

42-
func.func @cmpi_signed() {
42+
func.func @cmpi_eq() {
4343
// ------------------------------------------------
4444
// Test i1
4545
// ------------------------------------------------
@@ -48,10 +48,42 @@ func.func @cmpi_signed() {
4848
%true_i1_n1 = arith.constant -1 : i1
4949

5050
// int values 1 and -1 represent the same bitvector
51+
5152
// CHECK-LABEL: @cmpi_eq_i1
5253
// CHECK-NEXT: 1
5354
func.call @cmpi_eq_i1(%true_i1, %true_i1_n1) : (i1, i1) -> ()
5455

56+
// CHECK-LABEL: @cmpi_eq_i1
57+
// CHECK-NEXT: 0
58+
func.call @cmpi_eq_i1(%false_i1, %true_i1) : (i1, i1) -> ()
59+
60+
// CHECK-LABEL: @cmpi_eq_i1
61+
// CHECK-NEXT: 0
62+
func.call @cmpi_eq_i1(%true_i1, %false_i1) : (i1, i1) -> ()
63+
64+
// CHECK-LABEL: @cmpi_eq_i1
65+
// CHECK-NEXT: 1
66+
func.call @cmpi_eq_i1(%true_i1, %true_i1) : (i1, i1) -> ()
67+
68+
// CHECK-LABEL: @cmpi_eq_i1
69+
// CHECK-NEXT: 1
70+
func.call @cmpi_eq_i1(%false_i1, %false_i1) : (i1, i1) -> ()
71+
72+
// ------------------------------------------------
73+
// TODO: Test i8, i16 etc..
74+
// ------------------------------------------------
75+
return
76+
}
77+
78+
func.func @cmpi_signed() {
79+
// ------------------------------------------------
80+
// Test i1
81+
// ------------------------------------------------
82+
%false_i1 = arith.constant 0 : i1
83+
%true_i1 = arith.constant 1 : i1
84+
%true_i1_n1 = arith.constant -1 : i1
85+
86+
// int values 1 and -1 represent the same bitvector
5587
// But, bitvector `1` is interpreted as int value -1 in signed comparison
5688

5789
// CHECK-LABEL: @cmpi_sge_i1
@@ -117,6 +149,7 @@ func.func @cmpi_unsigned() {
117149
}
118150

119151
func.func @entry() {
152+
func.call @cmpi_eq() : () -> ()
120153
func.call @cmpi_signed() : () -> ()
121154
func.call @cmpi_unsigned() : () -> ()
122155
return

0 commit comments

Comments
 (0)