xref: /llvm-project/mlir/test/Dialect/SparseTensor/codegen_to_llvm.mlir (revision dbe1be9aa4e010f8ed945e19ba93a1f927aade8e)
11c7ffe0cSAart Bik// RUN: mlir-opt %s --sparse-tensor-codegen --sparse-storage-specifier-to-llvm | FileCheck %s
21c7ffe0cSAart Bik
3*dbe1be9aSYinying Li#SparseVector = #sparse_tensor.encoding<{ map = (d0) -> (d0 : compressed) }>
41c7ffe0cSAart Bik
51c7ffe0cSAart Bik// CHECK-LABEL: func @sparse_nop(
61c7ffe0cSAart Bik//  CHECK-SAME: %[[A0:.*0]]: memref<?xindex>,
71c7ffe0cSAart Bik//  CHECK-SAME: %[[A1:.*1]]: memref<?xindex>,
81c7ffe0cSAart Bik//  CHECK-SAME: %[[A2:.*2]]: memref<?xf64>,
91c7ffe0cSAart Bik//  CHECK-SAME: %[[A3:.*3]]: !llvm.struct<(array<1 x i64>, array<3 x i64>)>)
101c7ffe0cSAart Bik//       CHECK: return %[[A0]], %[[A1]], %[[A2]], %[[A3]] :
111c7ffe0cSAart Bik//  CHECK-SAME: memref<?xindex>, memref<?xindex>, memref<?xf64>, !llvm.struct<(array<1 x i64>, array<3 x i64>)>
121c7ffe0cSAart Bikfunc.func @sparse_nop(%arg0: tensor<?xf64, #SparseVector>) -> tensor<?xf64, #SparseVector> {
131c7ffe0cSAart Bik  return %arg0 : tensor<?xf64, #SparseVector>
141c7ffe0cSAart Bik}
15