Skip to content

Commit 2505904

Browse files
authored
[mlir][sparse] fix typo in sparse_tensor.unpack op (#66287)
SparseVector->COO
1 parent 29f6c0f commit 2505904

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -131,9 +131,10 @@ def SparseTensor_UnpackOp : SparseTensor_Op<"unpack", [Pure, SameVariadicResultS
131131
// input COO format |1.1, 0.0, 0.0, 0.0|
132132
// of 3x4 matrix |0.0, 0.0, 2.2, 3.3|
133133
// |0.0, 0.0, 0.0, 0.0|
134-
%v, %p, %c, %v_len, %p_len, %c_len = sparse_tensor.unpack %sp : tensor<3x4xf64, #SparseVector>
135-
outs(%od, %op, %oi : tensor<3xf64>, tensor<2xindex>, tensor<3x2xindex>)
136-
-> tensor<3xf64>, (tensor<2xindex>, tensor<3x2xindex>), index, (index, index)
134+
%v, %p, %c, %v_len, %p_len, %c_len =
135+
sparse_tensor.unpack %sp : tensor<3x4xf64, #COO>
136+
outs(%od, %op, %oi : tensor<3xf64>, tensor<2xindex>, tensor<3x2xindex>)
137+
-> tensor<3xf64>, (tensor<2xindex>, tensor<3x2xindex>), index, (index, index)
137138
// %v = arith.constant dense<[ 1.1, 2.2, 3.3 ]> : tensor<3xf64>
138139
// %p = arith.constant dense<[ 0, 3 ]> : tensor<2xindex>
139140
// %c = arith.constant dense<[[0,0], [1,2], [1,3]]> : tensor<3x2xindex>

0 commit comments

Comments
 (0)