[Mlir-commits] [mlir] [mlir][sparse] introduce sparse_tensor.reorder_coo operation (PR #68827)
Yinying Li
llvmlistbot at llvm.org
Wed Oct 11 14:55:09 PDT 2023
================
@@ -809,6 +809,36 @@ def SparseTensor_SortOp : SparseTensor_Op<"sort">,
let hasVerifier = 1;
}
+def SparseTensor_ReorderCOOOp : SparseTensor_Op<"reorder_coo", [Pure]>,
+ Arguments<(ins AnyCOOSparseTensor: $input_coo,
+ SparseTensorSortKindAttr:$algorithm)>,
+ Results<(outs AnyCOOSparseTensor: $result_coo)> {
+ let summary = "Reorder the input COO such that it has the the same order as "
+ "the output COO";
+ let description = [{
+ sparse_tensor.reorder_coo reorder input COO to the same order as specified by
+ the output format. E.g., reorder an unordered COO into an ordered one.
+
+ The input and result COO tensor must have the same element type, position type and
+ coordinate type. At the moment, the operation also only support ordering
----------------
yinying-lisa-li wrote:
supports?
https://github.com/llvm/llvm-project/pull/68827
More information about the Mlir-commits
mailing list