diff options
| author | Yinying Li <yinyingli@google.com> | 2024-03-12 21:39:37 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-12 21:39:37 -0400 |
| commit | 88986d65e4ed1b2ddd6693ff3e70b84436af767a (patch) | |
| tree | 291a65ad6739c70143bedbe2f391889d58bc7c58 | |
| parent | 9d6c43b4aed117f53167e72749b31a943941345d (diff) | |
| download | llvm-88986d65e4ed1b2ddd6693ff3e70b84436af767a.zip llvm-88986d65e4ed1b2ddd6693ff3e70b84436af767a.tar.gz llvm-88986d65e4ed1b2ddd6693ff3e70b84436af767a.tar.bz2 | |
[mlir][sparse] Fix sparse_generate test (#85009)
std::uniform_int_distribution may behave differently in different
systems.
| -rw-r--r-- | mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_generate.mlir | 17 |
1 files changed, 5 insertions, 12 deletions
diff --git a/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_generate.mlir b/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_generate.mlir index e1f73eb..63a6d3a 100644 --- a/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_generate.mlir +++ b/mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_generate.mlir @@ -78,20 +78,13 @@ module { } %sv = sparse_tensor.convert %output : tensor<?xf64> to tensor<?xf64, #SparseVector> + %n0 = sparse_tensor.number_of_entries %sv : tensor<?xf64, #SparseVector> + // Print the number of non-zeros for verification + // as shuffle may generate different numbers. // - // Verify the outputs. - // - // CHECK: ---- Sparse Tensor ---- - // CHECK-NEXT: nse = 5 - // CHECK-NEXT: dim = ( 50 ) - // CHECK-NEXT: lvl = ( 50 ) - // CHECK-NEXT: pos[0] : ( 0, 5 - // CHECK-NEXT: crd[0] : ( 1, 9, 17, 27, 30 - // CHECK-NEXT: values : ( 84, 34, 8, 40, 93 - // CHECK-NEXT: ---- - // - sparse_tensor.print %sv : tensor<?xf64, #SparseVector> + // CHECK: 5 + vector.print %n0 : index // Release the resources. bufferization.dealloc_tensor %sv : tensor<?xf64, #SparseVector> |