Skip to content

Commit 1f48f4d

Browse files
committed
[mlir][sparse] Fix a test to check all output coordinates.
Reviewed By: Peiming Differential Revision: https://reviews.llvm.org/D137805
1 parent 353f265 commit 1f48f4d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sorted_coo.mlir

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ module {
134134
// CHECK-NEXT: ( 0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
135135
// CHECK-NEXT: ( 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0 )
136136
// CHECK-NEXT: ( 0, 0, 1, 1, 2, 2, 2, 2, 0, 0, 0, 1, 1, 1, 1, 2, 2, 0, 0, 0 )
137-
// CHECK-NEXT: ( 0, 0, 1, 1, 2, 2, 2, 2, 0, 0, 0, 1, 1, 1, 1, 2, 2, 0, 0, 0 )
137+
// CHECK-NEXT: ( 2, 3, 1, 2, 0, 1, 2, 3, 0, 2, 3, 0, 1, 2, 3, 1, 2, 0, 0, 0 )
138138
// CHECK-NEXT: ( 3, 63, 11, 100, 66, 61, 13, 43, 77, 10, 46, 61, 53, 3, 75, 22, 18, nan, nan, nan )
139139
//
140140
%p2 = sparse_tensor.pointers %2 { dimension = 0 : index }
@@ -150,14 +150,14 @@ module {
150150
call @dumpi(%p2) : (memref<?xindex>) -> ()
151151
call @dumpi(%i20) : (memref<?xindex>) -> ()
152152
call @dumpi(%i21) : (memref<?xindex>) -> ()
153-
call @dumpi(%i21) : (memref<?xindex>) -> ()
153+
call @dumpi(%i22) : (memref<?xindex>) -> ()
154154
call @dumpf(%v2) : (memref<?xf64>) -> ()
155155

156156
//
157157
// CHECK-NEXT: ( 0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
158158
// CHECK-NEXT: ( 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 0, 0, 0 )
159159
// CHECK-NEXT: ( 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0 )
160-
// CHECK-NEXT: ( 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0 )
160+
// CHECK-NEXT: ( 2, 0, 1, 1, 2, 1, 2, 0, 1, 2, 0, 1, 2, 0, 2, 0, 1, 0, 0, 0 )
161161
// CHECK-NEXT: ( 66, 77, 61, 11, 61, 53, 22, 3, 100, 13, 10, 3, 18, 63, 43, 46, 75, nan, nan, nan )
162162
//
163163
%p3 = sparse_tensor.pointers %3 { dimension = 0 : index }
@@ -173,7 +173,7 @@ module {
173173
call @dumpi(%p3) : (memref<?xindex>) -> ()
174174
call @dumpi(%i30) : (memref<?xindex>) -> ()
175175
call @dumpi(%i31) : (memref<?xindex>) -> ()
176-
call @dumpi(%i31) : (memref<?xindex>) -> ()
176+
call @dumpi(%i32) : (memref<?xindex>) -> ()
177177
call @dumpf(%v3) : (memref<?xf64>) -> ()
178178

179179
//

0 commit comments

Comments
 (0)