[llvm] [mlir][sparse] implementating stageSparseOpPass as an interface (PR #69022)

Yinying Li via llvm-commits llvm-commits at lists.llvm.org
Mon Oct 16 10:14:13 PDT 2023


================
@@ -1248,6 +1248,23 @@ LogicalResult UnaryOp::verify() {
   return success();
 }
 
+bool ConcatenateOp::needExtraSort() {
+  SparseTensorType dstStt = getSparseTensorType(*this);
+  if (dstStt.isAllDense() || !dstStt.isAllOrdered())
+    return false;
+
+  bool allSameOrdered = llvm::all_of(getInputs(), [dstStt](Value op) {
+    return getSparseTensorType(op).hasSameDimToLvl(dstStt);
+  });
+  // TODO: When conDim != 0, as long as conDim corresponding to  the first level
----------------
yinying-lisa-li wrote:

nit: extra space between "to" and "the"

https://github.com/llvm/llvm-project/pull/69022


More information about the llvm-commits mailing list