Skip to content

Commit 1c7ffe0

Browse files
committed
[mlir][sparse] add test that combines sparse codegen and lowering to llvm struct
Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D141006
1 parent fb32a69 commit 1c7ffe0

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
// RUN: mlir-opt %s --sparse-tensor-codegen --sparse-storage-specifier-to-llvm | FileCheck %s
2+
3+
#SparseVector = #sparse_tensor.encoding<{ dimLevelType = ["compressed"] }>
4+
5+
// CHECK-LABEL: func @sparse_nop(
6+
// CHECK-SAME: %[[A0:.*0]]: memref<?xindex>,
7+
// CHECK-SAME: %[[A1:.*1]]: memref<?xindex>,
8+
// CHECK-SAME: %[[A2:.*2]]: memref<?xf64>,
9+
// CHECK-SAME: %[[A3:.*3]]: !llvm.struct<(array<1 x i64>, array<3 x i64>)>)
10+
// CHECK: return %[[A0]], %[[A1]], %[[A2]], %[[A3]] :
11+
// CHECK-SAME: memref<?xindex>, memref<?xindex>, memref<?xf64>, !llvm.struct<(array<1 x i64>, array<3 x i64>)>
12+
func.func @sparse_nop(%arg0: tensor<?xf64, #SparseVector>) -> tensor<?xf64, #SparseVector> {
13+
return %arg0 : tensor<?xf64, #SparseVector>
14+
}

0 commit comments

Comments
 (0)