From d4e16171e889df1be22cb9cc64c58c9966124db3 Mon Sep 17 00:00:00 2001
From: Aart Bik <ajcbik@google.com>
Date: Fri, 17 Sep 2021 11:55:45 -0700
Subject: [PATCH] [mlir][sparse] add dce test for all sparse tensor ops

Reviewed By: ThomasRaoux

Differential Revision: https://reviews.llvm.org/D109992
---
 mlir/test/Dialect/SparseTensor/fold.mlir | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/mlir/test/Dialect/SparseTensor/fold.mlir b/mlir/test/Dialect/SparseTensor/fold.mlir
index b26f003..035cac4 100644
--- a/mlir/test/Dialect/SparseTensor/fold.mlir
+++ b/mlir/test/Dialect/SparseTensor/fold.mlir
@@ -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
+}
-- 
2.7.4