]> https://gitweb.dealii.org/ - dealii.git/commitdiff
Test for Trilinos sparsity pattern initialization 1231/head
authorMartin Kronbichler <kronbichler@lnm.mw.tum.de>
Sat, 1 Aug 2015 20:10:29 +0000 (22:10 +0200)
committerMartin Kronbichler <kronbichler@lnm.mw.tum.de>
Sun, 2 Aug 2015 07:13:39 +0000 (09:13 +0200)
tests/trilinos/trilinos_sparsity_pattern_04.cc [new file with mode: 0644]
tests/trilinos/trilinos_sparsity_pattern_04.mpirun=3.output [new file with mode: 0644]

diff --git a/tests/trilinos/trilinos_sparsity_pattern_04.cc b/tests/trilinos/trilinos_sparsity_pattern_04.cc
new file mode 100644 (file)
index 0000000..e184f49
--- /dev/null
@@ -0,0 +1,219 @@
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2014 - 2015 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE at
+// the top level of the deal.II distribution.
+//
+// ---------------------------------------------------------------------
+
+
+
+// test several reinitialization options with a Trilinos sparsity pattern and
+// Trilinos sparse matrix
+
+#include "../tests.h"
+#include <deal.II/lac/trilinos_sparsity_pattern.h>
+#include <deal.II/lac/trilinos_sparse_matrix.h>
+#include <deal.II/lac/dynamic_sparsity_pattern.h>
+#include <fstream>
+#include <iomanip>
+
+
+void test ()
+{
+  const unsigned int n_procs = Utilities::MPI::n_mpi_processes(MPI_COMM_WORLD);
+  const unsigned int myid = Utilities::MPI::this_mpi_process(MPI_COMM_WORLD);
+
+  TrilinosWrappers::SparsityPattern sp;
+
+  deallog << "Creating entries..." << std::endl;
+
+  IndexSet rows(2*n_procs);
+  rows.add_range(2*myid, 2*myid+2);
+  rows.compress();
+  IndexSet writable_rows(rows);
+  writable_rows.add_index(1);
+  IndexSet columns(3*n_procs);
+  columns.add_range(3*myid, 3*myid+3);
+  columns.compress();
+
+  // this creates a matrix with optimized path for off-processor entries
+  sp.reinit(rows, columns, writable_rows, MPI_COMM_WORLD, 0);
+
+  for (unsigned int i=2*myid; i<2*myid+2; ++i)
+    for (unsigned int j=0; j<3*n_procs; ++j)
+      if ((i+2*j+1) % 3 == 0)
+        sp.add (i,j);
+
+  sp.add(1, 0);
+  sp.add(1, 4);
+
+  sp.compress ();
+
+  deallog << "SP::is_compressed(): " << sp.is_compressed() << std::endl;
+  deallog << "Number of entries: " << sp.n_nonzero_elements() << std::endl;
+  deallog << "Number of rows: " << sp.n_rows() << std::endl;
+  deallog << "Number of colums: " << sp.n_cols() << std::endl;
+
+  {
+    // create matrix from sparsity pattern, add a few entries and output the
+    // matrix norm
+    TrilinosWrappers::SparseMatrix matrix;
+    matrix.reinit(sp);
+    double c = 0;
+    for (unsigned int i=2*myid; i<2*myid+2; ++i)
+      for (unsigned int j=0; j<3*n_procs; ++j)
+        if ((i+2*j+1) % 3 == 0)
+          matrix.add (i,j,c++);
+
+    matrix.add(1, 0, c++);
+    matrix.add(1, 4, c++);
+
+    matrix.compress(VectorOperation::add);
+
+    deallog << "Matrix norm: " << matrix.frobenius_norm() << std::endl;
+  }
+
+  // reinit, this time without giving writable rows -> Trilinos must manage
+  // ghost entries and throw away the off-processor sparsity pattern
+  sp.reinit(rows, columns, MPI_COMM_WORLD, 0);
+
+  for (unsigned int i=2*myid; i<2*myid+2; ++i)
+    for (unsigned int j=0; j<3*n_procs; ++j)
+      if ((i+2*j+1) % 3 == 0)
+        sp.add (i,j);
+
+  sp.add(1, 0);
+  sp.add(1, 4);
+  sp.add(2, 1);
+  sp.add(2, 5);
+
+  sp.compress ();
+  {
+    // create matrix from sparsity pattern, add a few entries and output the
+    // matrix norm
+    TrilinosWrappers::SparseMatrix matrix;
+    matrix.reinit(sp);
+    double c = 0;
+    for (unsigned int i=2*myid; i<2*myid+2; ++i)
+      for (unsigned int j=0; j<3*n_procs; ++j)
+        if ((i+2*j+1) % 3 == 0)
+          matrix.add (i,j,c++);
+
+    matrix.add(1, 0, c++);
+    matrix.add(1, 4, c++);
+    matrix.add(2, 1, c++);
+    matrix.add(2, 5, c++);
+
+    matrix.compress(VectorOperation::add);
+
+    deallog << "Matrix norm: " << matrix.frobenius_norm() << std::endl;
+  }
+
+
+  // now create again a pattern with writable rows
+  sp.reinit(rows, columns, writable_rows, MPI_COMM_WORLD, 0);
+
+  for (unsigned int i=2*myid; i<2*myid+2; ++i)
+    for (unsigned int j=0; j<3*n_procs; ++j)
+      if ((i+2*j+1) % 3 == 0)
+        sp.add (i,j);
+
+  sp.add(1, 0);
+  sp.add(1, 4);
+
+  sp.compress ();
+
+  {
+    // create matrix from sparsity pattern, add a few entries and output the
+    // matrix norm
+    TrilinosWrappers::SparseMatrix matrix;
+    matrix.reinit(sp);
+    double c = 0;
+    for (unsigned int i=2*myid; i<2*myid+2; ++i)
+      for (unsigned int j=0; j<3*n_procs; ++j)
+        if ((i+2*j+1) % 3 == 0)
+          matrix.add (i,j,c++);
+
+    matrix.add(1, 0, c++);
+    matrix.add(1, 4, c++);
+
+    matrix.compress(VectorOperation::add);
+
+    deallog << "Matrix norm: " << matrix.frobenius_norm() << std::endl;
+  }
+
+
+  // finally, initialize the Trilinos sparsity pattern from a
+  // DynamicSparsityPattern
+  writable_rows.add_index(2);
+  writable_rows.compress();
+  DynamicSparsityPattern dsp(rows.size(), columns.size(), writable_rows);
+  for (unsigned int i=2*myid; i<2*myid+2; ++i)
+    for (unsigned int j=0; j<3*n_procs; ++j)
+      if ((i+2*j+1) % 3 == 0)
+        dsp.add (i,j);
+
+  dsp.add(1, 0);
+  dsp.add(1, 4);
+  dsp.add(2, 1);
+  dsp.add(2, 5);
+
+  sp.reinit(rows, columns, dsp, MPI_COMM_WORLD, true);
+  sp.compress ();
+  {
+    // create matrix from sparsity pattern, add a few entries and output the
+    // matrix norm
+    TrilinosWrappers::SparseMatrix matrix;
+    matrix.reinit(sp);
+    double c = 0;
+    for (unsigned int i=2*myid; i<2*myid+2; ++i)
+      for (unsigned int j=0; j<3*n_procs; ++j)
+        if ((i+2*j+1) % 3 == 0)
+          matrix.add (i,j,c++);
+
+    matrix.add(1, 0, c++);
+    matrix.add(1, 4, c++);
+    matrix.add(2, 1, c++);
+    matrix.add(2, 5, c++);
+
+    matrix.compress(VectorOperation::add);
+
+    deallog << "Matrix norm: " << matrix.frobenius_norm() << std::endl;
+  }
+
+  deallog << "OK" << std::endl;
+}
+
+
+
+int main (int argc, char **argv)
+{
+  std::ofstream logfile("output");
+  deallog.attach(logfile);
+  deallog.depth_console(0);
+  deallog.threshold_double(1.e-10);
+
+  Utilities::MPI::MPI_InitFinalize mpi_initialization (argc, argv, testing_max_num_threads());
+
+  if (Utilities::MPI::this_mpi_process(MPI_COMM_WORLD) == 0)
+    {
+      std::ofstream logfile ("output");
+      deallog.attach (logfile);
+      deallog.depth_console (0);
+      deallog.threshold_double (1.e-10);
+
+      test();
+    }
+  else
+    {
+      test();
+    }
+}
diff --git a/tests/trilinos/trilinos_sparsity_pattern_04.mpirun=3.output b/tests/trilinos/trilinos_sparsity_pattern_04.mpirun=3.output
new file mode 100644 (file)
index 0000000..4d3adf7
--- /dev/null
@@ -0,0 +1,11 @@
+
+DEAL::Creating entries...
+DEAL::SP::is_compressed(): 1
+DEAL::Number of entries: 20
+DEAL::Number of rows: 6
+DEAL::Number of colums: 9
+DEAL::Matrix norm: 30.4959
+DEAL::Matrix norm: 47.2758
+DEAL::Matrix norm: 30.4959
+DEAL::Matrix norm: 47.2758
+DEAL::OK

In the beginning the Universe was created. This has made a lot of people very angry and has been widely regarded as a bad move.

Douglas Adams


Typeset in Trocchi and Trocchi Bold Sans Serif.