write full matrix

LLNL · Sep 10, 2024 · 155bd5a · 155bd5a
1 parent 5e230da
commit 155bd5a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/apps/EDGE3D.hpp b/src/apps/EDGE3D.hpp
@@ -534,6 +534,19 @@ RAJA_INLINE void symmetric_edge_MpSmatrix(
       }
     }
   }
+
+  // write back to matrix
+  rajaperf::Int_type offset = 0;
+  for (rajaperf::Int_type p = 0; p < M; p++) {
+
+    matrix[p][p] = symmetric_matrix[offset];
+    for (rajaperf::Int_type m = 1; m < (M-p); m++) {
+      auto x = symmetric_matrix[offset + m];
+      matrix[p][m] = x;
+      matrix[m][p] = x;
+    }
+    offset += (M-p);
+  }
 }
 
 #define EDGE3D_DATA_SETUP \