[Mlir-commits] [mlir] [mlir][sparse] Change sparse_tensor.print format (PR #91528)

Aart Bik llvmlistbot at llvm.org
Wed May 8 14:09:24 PDT 2024


================
@@ -830,10 +830,12 @@ struct PrintRewriter : public OpRewritePattern<PrintOp> {
                                          vector::PrintPunctuation::Comma);
         rewriter.create<vector::PrintOp>(loc, imag,
                                          vector::PrintPunctuation::Close);
-        rewriter.create<vector::PrintOp>(loc, vector::PrintPunctuation::Comma);
+        printEnding(rewriter, loc, forOp.getInductionVar(), step,
+                    forOp.getUpperBound(), /*isComplex*/ true, val);
+
       } else {
-        rewriter.create<vector::PrintOp>(loc, val,
-                                         vector::PrintPunctuation::Comma);
+        printEnding(rewriter, loc, forOp.getInductionVar(), step,
----------------
aartbik wrote:

how about idxs.back() ?

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


More information about the Mlir-commits mailing list