This is an archive of the discontinued LLVM Phabricator instance.

Differential D137805

[mlir][sparse] Fix a test to check all output coordinates.
ClosedPublic

Authored by bixia on Nov 10 2022, 3:02 PM.

Details

Reviewers
aartbik
nicolasvasilache
Peiming
Commits
rG1f48f4d674d5: [mlir][sparse] Fix a test to check all output coordinates.

Diff Detail

Event Timeline

bixia created this revision.Nov 10 2022, 3:02 PM
Herald added a reviewer: aartbik. · View Herald TranscriptNov 10 2022, 3:02 PM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: Moerafaat, anlunx, bzcheeseman and 21 others. · View Herald Transcript
bixia requested review of this revision.Nov 10 2022, 3:02 PM
Herald added a reviewer: nicolasvasilache. · View Herald TranscriptNov 10 2022, 3:02 PM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
bixia added a reviewer: Peiming.Nov 10 2022, 3:02 PM
Peiming accepted this revision.Nov 10 2022, 3:05 PM
This revision is now accepted and ready to land.Nov 10 2022, 3:05 PM
Harbormaster completed remote builds in B197129: Diff 474617.Nov 10 2022, 3:19 PM
Closed by commit rG1f48f4d674d5: [mlir][sparse] Fix a test to check all output coordinates. (authored by bixia). · Explain WhyNov 10 2022, 3:26 PM
This revision was automatically updated to reflect the committed changes.
bixia added a commit: rG1f48f4d674d5: [mlir][sparse] Fix a test to check all output coordinates..

Revision Contents

PathSize
mlir/
test/
Integration/
Dialect/
SparseTensor/
CPU/
8 lines

Diff 474621

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

Show First 20 Lines • Show All 128 Lines • ▼ Show 20 Lines func.func @entry() {
call @dumpi(%i10) : (memref<?xindex>) -> () call @dumpi(%i10) : (memref<?xindex>) -> ()
call @dumpi(%i11) : (memref<?xindex>) -> () call @dumpi(%i11) : (memref<?xindex>) -> ()
call @dumpf(%v1) : (memref<?xf64>) -> () call @dumpf(%v1) : (memref<?xf64>) -> ()
// //
// CHECK-NEXT: ( 0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ) // CHECK-NEXT: ( 0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
// CHECK-NEXT: ( 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0 ) // CHECK-NEXT: ( 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0 )
// CHECK-NEXT: ( 0, 0, 1, 1, 2, 2, 2, 2, 0, 0, 0, 1, 1, 1, 1, 2, 2, 0, 0, 0 ) // CHECK-NEXT: ( 0, 0, 1, 1, 2, 2, 2, 2, 0, 0, 0, 1, 1, 1, 1, 2, 2, 0, 0, 0 )
// CHECK-NEXT: ( 0, 0, 1, 1, 2, 2, 2, 2, 0, 0, 0, 1, 1, 1, 1, 2, 2, 0, 0, 0 ) // CHECK-NEXT: ( 2, 3, 1, 2, 0, 1, 2, 3, 0, 2, 3, 0, 1, 2, 3, 1, 2, 0, 0, 0 )
// CHECK-NEXT: ( 3, 63, 11, 100, 66, 61, 13, 43, 77, 10, 46, 61, 53, 3, 75, 22, 18, nan, nan, nan ) // CHECK-NEXT: ( 3, 63, 11, 100, 66, 61, 13, 43, 77, 10, 46, 61, 53, 3, 75, 22, 18, nan, nan, nan )
// //
%p2 = sparse_tensor.pointers %2 { dimension = 0 : index } %p2 = sparse_tensor.pointers %2 { dimension = 0 : index }
: tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex>
%i20 = sparse_tensor.indices %2 { dimension = 0 : index } %i20 = sparse_tensor.indices %2 { dimension = 0 : index }
: tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex>
%i21 = sparse_tensor.indices %2 { dimension = 1 : index } %i21 = sparse_tensor.indices %2 { dimension = 1 : index }
: tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex>
%i22 = sparse_tensor.indices %2 { dimension = 2 : index } %i22 = sparse_tensor.indices %2 { dimension = 2 : index }
: tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3D> to memref<?xindex>
%v2 = sparse_tensor.values %2 %v2 = sparse_tensor.values %2
: tensor<?x?x?xf64, #SortedCOO3D> to memref<?xf64> : tensor<?x?x?xf64, #SortedCOO3D> to memref<?xf64>
call @dumpi(%p2) : (memref<?xindex>) -> () call @dumpi(%p2) : (memref<?xindex>) -> ()
call @dumpi(%i20) : (memref<?xindex>) -> () call @dumpi(%i20) : (memref<?xindex>) -> ()
call @dumpi(%i21) : (memref<?xindex>) -> () call @dumpi(%i21) : (memref<?xindex>) -> ()
call @dumpi(%i21) : (memref<?xindex>) -> () call @dumpi(%i22) : (memref<?xindex>) -> ()
call @dumpf(%v2) : (memref<?xf64>) -> () call @dumpf(%v2) : (memref<?xf64>) -> ()
// //
// CHECK-NEXT: ( 0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ) // CHECK-NEXT: ( 0, 17, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
// CHECK-NEXT: ( 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 0, 0, 0 ) // CHECK-NEXT: ( 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 0, 0, 0 )
// CHECK-NEXT: ( 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0 ) // CHECK-NEXT: ( 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0 )
// CHECK-NEXT: ( 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0 ) // CHECK-NEXT: ( 2, 0, 1, 1, 2, 1, 2, 0, 1, 2, 0, 1, 2, 0, 2, 0, 1, 0, 0, 0 )
// CHECK-NEXT: ( 66, 77, 61, 11, 61, 53, 22, 3, 100, 13, 10, 3, 18, 63, 43, 46, 75, nan, nan, nan ) // CHECK-NEXT: ( 66, 77, 61, 11, 61, 53, 22, 3, 100, 13, 10, 3, 18, 63, 43, 46, 75, nan, nan, nan )
// //
%p3 = sparse_tensor.pointers %3 { dimension = 0 : index } %p3 = sparse_tensor.pointers %3 { dimension = 0 : index }
: tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex>
%i30 = sparse_tensor.indices %3 { dimension = 0 : index } %i30 = sparse_tensor.indices %3 { dimension = 0 : index }
: tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex>
%i31 = sparse_tensor.indices %3 { dimension = 1 : index } %i31 = sparse_tensor.indices %3 { dimension = 1 : index }
: tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex>
%i32 = sparse_tensor.indices %3 { dimension = 2 : index } %i32 = sparse_tensor.indices %3 { dimension = 2 : index }
: tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex> : tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xindex>
%v3 = sparse_tensor.values %3 %v3 = sparse_tensor.values %3
: tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xf64> : tensor<?x?x?xf64, #SortedCOO3DPermuted> to memref<?xf64>
call @dumpi(%p3) : (memref<?xindex>) -> () call @dumpi(%p3) : (memref<?xindex>) -> ()
call @dumpi(%i30) : (memref<?xindex>) -> () call @dumpi(%i30) : (memref<?xindex>) -> ()
call @dumpi(%i31) : (memref<?xindex>) -> () call @dumpi(%i31) : (memref<?xindex>) -> ()
call @dumpi(%i31) : (memref<?xindex>) -> () call @dumpi(%i32) : (memref<?xindex>) -> ()
call @dumpf(%v3) : (memref<?xf64>) -> () call @dumpf(%v3) : (memref<?xf64>) -> ()
// //
// CHECK-NEXT: ( 0, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ) // CHECK-NEXT: ( 0, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
// CHECK-NEXT: ( 0, 1, 2, 2, 3, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ) // CHECK-NEXT: ( 0, 1, 2, 2, 3, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
// CHECK-NEXT: ( 0, 3, 0, 3, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ) // CHECK-NEXT: ( 0, 3, 0, 3, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 )
// CHECK-NEXT: ( 6, 5, 4, 3, 2, 11, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan ) // CHECK-NEXT: ( 6, 5, 4, 3, 2, 11, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan )
// //
▲ Show 20 Lines • Show All 46 Lines • Show Last 20 Lines