maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

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

Reviewed By: ThomasRaoux

Differential Revision: https://reviews.llvm.org/D109992
This commit is contained in:
Aart Bik 2021-09-17 11:55:45 -07:00
parent d01e0c8c66
commit d4e16171e8
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
mlir/test/Dialect/SparseTensor/fold.mlir
View File

@ -1,5 +1,6 @@
// RUN: mlir-opt %s --canonicalize --cse | FileCheck %s
#DenseVector = #sparse_tensor.encoding<{dimLevelType = ["dense"]}>
#SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
// CHECK-LABEL: func @sparse_nop_convert(
@ -18,3 +19,28 @@ func @sparse_dce_convert(%arg0: tensor<64xf32>) {
%0 = sparse_tensor.convert %arg0 : tensor<64xf32> to tensor<64xf32, #SparseVector>
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
}