diff --git a/mlir/test/Dialect/SparseTensor/codegen_to_llvm.mlir b/mlir/test/Dialect/SparseTensor/codegen_to_llvm.mlir new file mode 100644 --- /dev/null +++ b/mlir/test/Dialect/SparseTensor/codegen_to_llvm.mlir @@ -0,0 +1,14 @@ +// RUN: mlir-opt %s --sparse-tensor-codegen --sparse-storage-specifier-to-llvm | FileCheck %s + +#SparseVector = #sparse_tensor.encoding<{ dimLevelType = ["compressed"] }> + +// CHECK-LABEL: func @sparse_nop( +// CHECK-SAME: %[[A0:.*0]]: memref, +// CHECK-SAME: %[[A1:.*1]]: memref, +// CHECK-SAME: %[[A2:.*2]]: memref, +// CHECK-SAME: %[[A3:.*3]]: !llvm.struct<(array<1 x i64>, array<3 x i64>)>) +// CHECK: return %[[A0]], %[[A1]], %[[A2]], %[[A3]] : +// CHECK-SAME: memref, memref, memref, !llvm.struct<(array<1 x i64>, array<3 x i64>)> +func.func @sparse_nop(%arg0: tensor) -> tensor { + return %arg0 : tensor +}