diff --git a/mlir/test/Dialect/SparseTensor/conversion.mlir b/mlir/test/Dialect/SparseTensor/conversion.mlir --- a/mlir/test/Dialect/SparseTensor/conversion.mlir +++ b/mlir/test/Dialect/SparseTensor/conversion.mlir @@ -1,10 +1,4 @@ -// First use with `kViaCOO` for sparse2sparse conversion (the old way). -// RUN: mlir-opt %s --sparse-tensor-conversion="s2s-strategy=1" \ -// RUN: --canonicalize --cse | FileCheck %s -// -// Now again with `kAuto` (the new default). -// RUN: mlir-opt %s --sparse-tensor-conversion="s2s-strategy=0" \ -// RUN: --canonicalize --cse | FileCheck %s -check-prefix=CHECKAUTO +// RUN: mlir-opt %s --sparse-tensor-conversion --canonicalize --cse | FileCheck %s #SparseVector = #sparse_tensor.encoding<{ dimLevelType = ["compressed"] @@ -233,29 +227,15 @@ // CHECK-LABEL: func @sparse_convert_1d_ss( // CHECK-SAME: %[[A:.*]]: !llvm.ptr) -// CHECK-DAG: %[[ToCOO:.*]] = arith.constant 5 : i32 -// CHECK-DAG: %[[FromCOO:.*]] = arith.constant 2 : i32 +// CHECK-DAG: %[[SparseToSparse:.*]] = arith.constant 3 : i32 // CHECK-DAG: %[[P:.*]] = memref.alloca() : memref<1xi8> // CHECK-DAG: %[[Q:.*]] = memref.alloca() : memref<1xindex> // CHECK-DAG: %[[R:.*]] = memref.alloca() : memref<1xindex> // CHECK-DAG: %[[X:.*]] = memref.cast %[[P]] : memref<1xi8> to memref // CHECK-DAG: %[[Y:.*]] = memref.cast %[[Q]] : memref<1xindex> to memref // CHECK-DAG: %[[Z:.*]] = memref.cast %[[R]] : memref<1xindex> to memref -// CHECK: %[[C:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[ToCOO]], %[[A]]) -// CHECK: %[[T:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[FromCOO]], %[[C]]) -// CHECK: call @delSparseTensorCOOF32(%[[C]]) +// CHECK: %[[T:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[SparseToSparse]], %[[A]]) // CHECK: return %[[T]] : !llvm.ptr -// CHECKAUTO-LABEL: func @sparse_convert_1d_ss( -// CHECKAUTO-SAME: %[[A:.*]]: !llvm.ptr) -// CHECKAUTO-DAG: %[[SparseToSparse:.*]] = arith.constant 3 : i32 -// CHECKAUTO-DAG: %[[P:.*]] = memref.alloca() : memref<1xi8> -// CHECKAUTO-DAG: %[[Q:.*]] = memref.alloca() : memref<1xindex> -// CHECKAUTO-DAG: %[[R:.*]] = memref.alloca() : memref<1xindex> -// CHECKAUTO-DAG: %[[X:.*]] = memref.cast %[[P]] : memref<1xi8> to memref -// CHECKAUTO-DAG: %[[Y:.*]] = memref.cast %[[Q]] : memref<1xindex> to memref -// CHECKAUTO-DAG: %[[Z:.*]] = memref.cast %[[R]] : memref<1xindex> to memref -// CHECKAUTO: %[[T:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[SparseToSparse]], %[[A]]) -// CHECKAUTO: return %[[T]] : !llvm.ptr func.func @sparse_convert_1d_ss(%arg0: tensor) -> tensor { %0 = sparse_tensor.convert %arg0 : tensor to tensor return %0 : tensor diff --git a/mlir/test/Dialect/SparseTensor/conversion_sparse2sparse.mlir b/mlir/test/Dialect/SparseTensor/conversion_sparse2sparse.mlir new file mode 100644 --- /dev/null +++ b/mlir/test/Dialect/SparseTensor/conversion_sparse2sparse.mlir @@ -0,0 +1,49 @@ +// First use with `kViaCOO` for sparse2sparse conversion (the old way). +// RUN: mlir-opt %s --sparse-tensor-conversion="s2s-strategy=1" \ +// RUN: --canonicalize --cse | FileCheck %s -check-prefix=CHECK-COO +// +// Now again with `kAuto` (the new default). +// RUN: mlir-opt %s --sparse-tensor-conversion="s2s-strategy=0" \ +// RUN: --canonicalize --cse | FileCheck %s -check-prefix=CHECK-AUTO + +#SparseVector64 = #sparse_tensor.encoding<{ + dimLevelType = ["compressed"], + pointerBitWidth = 64, + indexBitWidth = 64 +}> + +#SparseVector32 = #sparse_tensor.encoding<{ + dimLevelType = ["compressed"], + pointerBitWidth = 32, + indexBitWidth = 32 +}> + +// CHECK-COO-LABEL: func @sparse_convert( +// CHECK-COO-SAME: %[[A:.*]]: !llvm.ptr) +// CHECK-COO-DAG: %[[ToCOO:.*]] = arith.constant 5 : i32 +// CHECK-COO-DAG: %[[FromCOO:.*]] = arith.constant 2 : i32 +// CHECK-COO-DAG: %[[P:.*]] = memref.alloca() : memref<1xi8> +// CHECK-COO-DAG: %[[Q:.*]] = memref.alloca() : memref<1xindex> +// CHECK-COO-DAG: %[[R:.*]] = memref.alloca() : memref<1xindex> +// CHECK-COO-DAG: %[[X:.*]] = memref.cast %[[P]] : memref<1xi8> to memref +// CHECK-COO-DAG: %[[Y:.*]] = memref.cast %[[Q]] : memref<1xindex> to memref +// CHECK-COO-DAG: %[[Z:.*]] = memref.cast %[[R]] : memref<1xindex> to memref +// CHECK-COO: %[[C:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[ToCOO]], %[[A]]) +// CHECK-COO: %[[T:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[FromCOO]], %[[C]]) +// CHECK-COO: call @delSparseTensorCOOF32(%[[C]]) +// CHECK-COO: return %[[T]] : !llvm.ptr +// CHECK-AUTO-LABEL: func @sparse_convert( +// CHECK-AUTO-SAME: %[[A:.*]]: !llvm.ptr) +// CHECK-AUTO-DAG: %[[SparseToSparse:.*]] = arith.constant 3 : i32 +// CHECK-AUTO-DAG: %[[P:.*]] = memref.alloca() : memref<1xi8> +// CHECK-AUTO-DAG: %[[Q:.*]] = memref.alloca() : memref<1xindex> +// CHECK-AUTO-DAG: %[[R:.*]] = memref.alloca() : memref<1xindex> +// CHECK-AUTO-DAG: %[[X:.*]] = memref.cast %[[P]] : memref<1xi8> to memref +// CHECK-AUTO-DAG: %[[Y:.*]] = memref.cast %[[Q]] : memref<1xindex> to memref +// CHECK-AUTO-DAG: %[[Z:.*]] = memref.cast %[[R]] : memref<1xindex> to memref +// CHECK-AUTO: %[[T:.*]] = call @newSparseTensor(%[[X]], %[[Y]], %[[Z]], %{{.*}}, %{{.*}}, %{{.*}}, %[[SparseToSparse]], %[[A]]) +// CHECK-AUTO: return %[[T]] : !llvm.ptr +func.func @sparse_convert(%arg0: tensor) -> tensor { + %0 = sparse_tensor.convert %arg0 : tensor to tensor + return %0 : tensor +}