from PR #253: bug fix to MMWrite for dense matrices

GraphBLAS · Jul 1, 2024 · 19d8f3f · 19d8f3f
1 parent 8270262
commit 19d8f3f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/data/comments_full.mtx b/data/comments_full.mtx
@@ -2,7 +2,7 @@
 %%GraphBLAS type double
 % comments for full.mtx
 % this file was created by test_MMRead.c
-3 3 9
+3 3
 .646
 .709
 .754

diff --git a/src/utility/LAGraph_MMWrite.c b/src/utility/LAGraph_MMWrite.c
@@ -439,8 +439,18 @@ int LAGraph_MMWrite
         nvals_to_print = nself_edges + (nvals - nself_edges) / 2 ;
     }
 
-    FPRINTF (f, "%" PRIu64 " %" PRIu64 " %" PRIu64 "\n",
-        nrows, ncols, nvals_to_print) ;
+    if (MM_fmt == MM_array)
+    {
+        // write `nrows ncols` if the array format is used
+        FPRINTF (f, "%" PRIu64 " %" PRIu64 "\n",
+            nrows, ncols) ;
+    }
+    else
+    {
+        // otherwise write `nrows ncols nvals` for the coordinate format
+        FPRINTF (f, "%" PRIu64 " %" PRIu64 " %" PRIu64 "\n",
+            nrows, ncols, nvals_to_print) ;
+    }
 
     if (nvals_to_print == 0)
     {