multifrontal_schur_gemm1

Source: tests/multifrontal/detail/multifrontal_schur_gemm.cpp

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

// Containers.
#include <myramath/sparse/Pattern.h>
#include <myramath/sparse/PatternRange.h>
#include <myramath/sparse/PatternBuilder.h>
#include <myramath/multifrontal/symbolic/AssemblyTree.h>
#include <myramath/multifrontal/symbolic/SchurTree.h>

// Algorithms.
#include <myramath/sparse/laplacian3.h>

// JobGraph under test.
#include <myramath/jobgraph/verify.h>
#include <myramath/jobgraph/graphviz.h>
#include <myramath/multifrontal/detail/schurgemm.h>

// Reporting.
#include <tests/myratest.h>

using namespace myra;
typedef multifrontal::Options Options;

ADD_TEST("multifrontal_schur_gemm1","[multifrontal][jobgraph]")
  {
  // Construct AssemblyTree of a 3D laplacian.
  int I = 32;
  int J = 32;
  int K = 32;
  int N = I*J*K;
  Pattern A_pattern = stencil3(I,J,K);
  Permutation perm = bisect3(I,J,K);
  AssemblyTree atree(A_pattern,perm);
  // Construct SchurTree for k=0 plane.
  Natural3D n(I,J,K);
  PatternBuilder B_builder(N,I*J);
  int b_next = 0;
  for (int i = 0; i < I; ++i)
    for (int j = 0; j < J; ++j)
      B_builder.insert(n(i,j,0),b_next++);
  Pattern B_pattern = B_builder.make_Pattern();
  SchurTree btree(atree,B_pattern);
  // Construct SchurTree for k=K-1 plane.
  PatternBuilder C_builder(N,I*J);
  int c_next = 0;
  for (int i = 0; i < I; ++i)
    for (int j = 0; j < J; ++j)
      C_builder.insert(n(i,j,K-1),c_next++);  
  Pattern C_pattern = C_builder.make_Pattern();
  SchurTree ctree(atree,C_pattern);
  // Verify the JobGraph.
  typedef ::multifrontal::detail::schurgemm::JobGraphBase1 Graph;  
  Graph graph(&atree,&btree,&ctree);
  //REQUIRE( verify(graph) );
  REQUIRE( verify(graph) );  
  // Save .dot file to disk?
  //graphviz(graph,"graph.dot");
  graphviz(graph,"graph2.dot");  
  }


ADD_TEST("multifrontal_schur_gemm2","[multifrontal]")
  {
  // Make diagonal Pattern for A.
  int N = 10;
  PatternBuilder A_builder(N,N);
  for (int n = 0; n < N; ++n)
    A_builder.insert(n,n);
  Pattern A_pattern = A_builder.make_Pattern();
  auto perm = Permutation::identity(N);
  typedef multifrontal::Options Options;
  Options options = Options::create().set_blocksize(1).set_globsize(1);
  AssemblyTree atree(A_pattern,perm,options);
  // Construct SchurTree for single-column SparseMatrix, containing [ 0 0 0 1 .. 1]
  PatternBuilder B_builder(N,1);
  for (int n = 3; n < N; n++)
    B_builder.insert(n,0);
  Pattern B_pattern = B_builder.make_Pattern();
  SchurTree btree(atree,B_pattern,options);
  // Construct SchurTree for single-column SparseMatrix, containing [ 1 .. 1 0 0 0 ]
  PatternBuilder C_builder(N,1);
  for (int n = 0; n < N-3; n++)
    C_builder.insert(n,0);
  Pattern C_pattern = C_builder.make_Pattern();
  SchurTree ctree(atree,C_pattern,options);
  // Verify the JobGraph.
  typedef ::multifrontal::detail::schurgemm::JobGraphBase1 Graph;  
  Graph graph(&atree,&btree,&ctree);
  REQUIRE( verify(graph) );  
  // Save .dot file to disk?
  graphviz(graph,"graph2.dot");  
  }

Results: [PASS]

Go back to Summary of /test programs.