Skip to content

Commit 88986d6

Browse files
[mlir][sparse] Fix sparse_generate test (#85009)
std::uniform_int_distribution may behave differently in different systems.
1 parent 9d6c43b commit 88986d6

File tree

1 file changed

+5
-12
lines changed

1 file changed

+5
-12
lines changed

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

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -78,20 +78,13 @@ module {
7878
}
7979

8080
%sv = sparse_tensor.convert %output : tensor<?xf64> to tensor<?xf64, #SparseVector>
81+
%n0 = sparse_tensor.number_of_entries %sv : tensor<?xf64, #SparseVector>
8182

83+
// Print the number of non-zeros for verification
84+
// as shuffle may generate different numbers.
8285
//
83-
// Verify the outputs.
84-
//
85-
// CHECK: ---- Sparse Tensor ----
86-
// CHECK-NEXT: nse = 5
87-
// CHECK-NEXT: dim = ( 50 )
88-
// CHECK-NEXT: lvl = ( 50 )
89-
// CHECK-NEXT: pos[0] : ( 0, 5
90-
// CHECK-NEXT: crd[0] : ( 1, 9, 17, 27, 30
91-
// CHECK-NEXT: values : ( 84, 34, 8, 40, 93
92-
// CHECK-NEXT: ----
93-
//
94-
sparse_tensor.print %sv : tensor<?xf64, #SparseVector>
86+
// CHECK: 5
87+
vector.print %n0 : index
9588

9689
// Release the resources.
9790
bufferization.dealloc_tensor %sv : tensor<?xf64, #SparseVector>

0 commit comments

Comments
 (0)