updated tester

ginkgo-project · Dec 9, 2023 · 20cd1f3 · 20cd1f3
1 parent 6aed992
commit 20cd1f3
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/examples/reordered-preconditioned-solver/reordered-preconditioned-solver.cpp b/examples/reordered-preconditioned-solver/reordered-preconditioned-solver.cpp
@@ -25,14 +25,13 @@ int main(int argc, char* argv[])
     auto A_data = gko::read_raw<ValueType, IndexType>(stream);
     gko::matrix_data<ValueType, IndexType> repeated_data{
         gko::dim<2>{A_data.size[0] * 3, A_data.size[0] * 3}};
-    repeated_data.nonzeros = A_data.nonzeros;
     for (auto entry : A_data.nonzeros) {
-        repeated_data.nonzeros.emplace_back(entry.row + A_data.size[0],
-                                            entry.column + A_data.size[0],
-                                            entry.value);
-        repeated_data.nonzeros.emplace_back(entry.row + 2 * A_data.size[0],
-                                            entry.column + 2 * A_data.size[0],
+        repeated_data.nonzeros.emplace_back(3 * entry.row, 3 * entry.column,
                                             entry.value);
+        repeated_data.nonzeros.emplace_back(3 * entry.row + 1,
+                                            3 * entry.column + 1, entry.value);
+        repeated_data.nonzeros.emplace_back(3 * entry.row + 2,
+                                            3 * entry.column + 2, entry.value);
     }
     repeated_data.sort_row_major();
     auto A = share(mtx::create(exec));