This is an archive of the discontinued LLVM Phabricator instance.

Differential D109992

[mlir][sparse] add dce test for all sparse tensor ops
ClosedPublic

Authored by aartbik on Sep 17 2021, 12:03 PM.

Details

Reviewers
ThomasRaoux
bixia
wrengr
Commits
rGd4e16171e889: [mlir][sparse] add dce test for all sparse tensor ops

Diff Detail

Event Timeline

aartbik created this revision.Sep 17 2021, 12:03 PM
Herald added subscribers: wenzhicui, wrengr, Chia-hungDuan and 18 others. · View Herald TranscriptSep 17 2021, 12:03 PM
aartbik requested review of this revision.Sep 17 2021, 12:03 PM
Herald added a project: Restricted Project. · View Herald TranscriptSep 17 2021, 12:03 PM
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
aartbik added reviewers: ThomasRaoux, bixia, wrengr.Sep 17 2021, 12:04 PM
ThomasRaoux accepted this revision.Sep 17 2021, 12:05 PM
This revision is now accepted and ready to land.Sep 17 2021, 12:05 PM
Harbormaster completed remote builds in B124453: Diff 373299.Sep 17 2021, 12:36 PM
Closed by commit rGd4e16171e889: [mlir][sparse] add dce test for all sparse tensor ops (authored by aartbik). · Explain WhySep 17 2021, 1:03 PM
This revision was automatically updated to reflect the committed changes.
aartbik added a commit: rGd4e16171e889: [mlir][sparse] add dce test for all sparse tensor ops.

Revision Contents

PathSize
mlir/
test/
Dialect/
SparseTensor/
26 lines

Diff 373320

mlir/test/Dialect/SparseTensor/fold.mlir

// RUN: mlir-opt %s --canonicalize --cse | FileCheck %s // RUN: mlir-opt %s --canonicalize --cse | FileCheck %s
#DenseVector = #sparse_tensor.encoding<{dimLevelType = ["dense"]}>
#SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}> #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
// CHECK-LABEL: func @sparse_nop_convert( // CHECK-LABEL: func @sparse_nop_convert(
// CHECK-SAME: %[[A:.*]]: tensor<64xf32, #{{.*}}>) // CHECK-SAME: %[[A:.*]]: tensor<64xf32, #{{.*}}>)
// CHECK: return %[[A]] : tensor<64xf32, #{{.*}}> // CHECK: return %[[A]] : tensor<64xf32, #{{.*}}>
func @sparse_nop_convert(%arg0: tensor<64xf32, #SparseVector>) -> tensor<64xf32, #SparseVector> { func @sparse_nop_convert(%arg0: tensor<64xf32, #SparseVector>) -> tensor<64xf32, #SparseVector> {
%0 = sparse_tensor.convert %arg0 : tensor<64xf32, #SparseVector> to tensor<64xf32, #SparseVector> %0 = sparse_tensor.convert %arg0 : tensor<64xf32, #SparseVector> to tensor<64xf32, #SparseVector>
return %0 : tensor<64xf32, #SparseVector> return %0 : tensor<64xf32, #SparseVector>
} }
// CHECK-LABEL: func @sparse_dce_convert( // CHECK-LABEL: func @sparse_dce_convert(
// CHECK-SAME: %[[A:.*]]: tensor<64xf32>) // CHECK-SAME: %[[A:.*]]: tensor<64xf32>)
// CHECK-NOT: sparse_tensor.convert // CHECK-NOT: sparse_tensor.convert
// CHECK: return // CHECK: return
func @sparse_dce_convert(%arg0: tensor<64xf32>) { func @sparse_dce_convert(%arg0: tensor<64xf32>) {
%0 = sparse_tensor.convert %arg0 : tensor<64xf32> to tensor<64xf32, #SparseVector> %0 = sparse_tensor.convert %arg0 : tensor<64xf32> to tensor<64xf32, #SparseVector>
return return
} }
// CHECK-LABEL: func @sparse_dce_getters(
// CHECK-SAME: %[[A:.*]]: tensor<64xf32, #sparse_tensor.encoding<{{{.*}}}>>)
// CHECK-NOT: sparse_tensor.pointers
// CHECK-NOT: sparse_tensor.indices
// CHECK-NOT: sparse_tensor.values
// CHECK: return
func @sparse_dce_getters(%arg0: tensor<64xf32, #SparseVector>) {
%c = constant 0 : index
%0 = sparse_tensor.pointers %arg0, %c : tensor<64xf32, #SparseVector> to memref<?xindex>
%1 = sparse_tensor.indices %arg0, %c : tensor<64xf32, #SparseVector> to memref<?xindex>
%2 = sparse_tensor.values %arg0 : tensor<64xf32, #SparseVector> to memref<?xf32>
return
}
// CHECK-LABEL: func @sparse_dce_reconstruct(
// CHECK-SAME: %[[A:.*]]: tensor<64xf32, #sparse_tensor.encoding<{{{.*}}}>>)
// CHECK-NOT: sparse_tensor.values
// CHECK-NOT: sparse_tensor.tensor
// CHECK: return
func @sparse_dce_reconstruct(%arg0: tensor<64xf32, #DenseVector>) {
%0 = sparse_tensor.values %arg0 : tensor<64xf32, #DenseVector> to memref<?xf32>
%1 = sparse_tensor.tensor %0 : memref<?xf32> to tensor<64xf32, #DenseVector>
return
}