diff options
| author | Aart Bik <39774503+aartbik@users.noreply.github.com> | 2023-09-13 14:51:03 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-13 14:51:03 -0700 |
| commit | 2505904901be8e584f6f3c68d59b93de296e95a0 (patch) | |
| tree | b1365863de0e56cbfefa6cc6b19041a273d7057c | |
| parent | 29f6c0fdfeb48f5f766d2ba02014358df69e8170 (diff) | |
| download | llvm-2505904901be8e584f6f3c68d59b93de296e95a0.zip llvm-2505904901be8e584f6f3c68d59b93de296e95a0.tar.gz llvm-2505904901be8e584f6f3c68d59b93de296e95a0.tar.bz2 | |
[mlir][sparse] fix typo in sparse_tensor.unpack op (#66287)
SparseVector->COO
| -rw-r--r-- | mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td index 7430a3c..94301db 100644 --- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td +++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td @@ -131,9 +131,10 @@ def SparseTensor_UnpackOp : SparseTensor_Op<"unpack", [Pure, SameVariadicResultS // input COO format |1.1, 0.0, 0.0, 0.0| // of 3x4 matrix |0.0, 0.0, 2.2, 3.3| // |0.0, 0.0, 0.0, 0.0| - %v, %p, %c, %v_len, %p_len, %c_len = sparse_tensor.unpack %sp : tensor<3x4xf64, #SparseVector> - outs(%od, %op, %oi : tensor<3xf64>, tensor<2xindex>, tensor<3x2xindex>) - -> tensor<3xf64>, (tensor<2xindex>, tensor<3x2xindex>), index, (index, index) + %v, %p, %c, %v_len, %p_len, %c_len = + sparse_tensor.unpack %sp : tensor<3x4xf64, #COO> + outs(%od, %op, %oi : tensor<3xf64>, tensor<2xindex>, tensor<3x2xindex>) + -> tensor<3xf64>, (tensor<2xindex>, tensor<3x2xindex>), index, (index, index) // %v = arith.constant dense<[ 1.1, 2.2, 3.3 ]> : tensor<3xf64> // %p = arith.constant dense<[ 0, 3 ]> : tensor<2xindex> // %c = arith.constant dense<[[0,0], [1,2], [1,3]]> : tensor<3x2xindex> |