20 ColorerTest(
const Teuchos::RCP<
const Teuchos::Comm<int> > &comm,
int multiple)
22 int me = comm->getRank();
23 int np = comm->getSize();
27 Teuchos::Array<gno_t> myRows(myNrows);
28 for (
size_t i = 0; i < myNrows; i++) {
29 myRows[i] = multiple * (me * myNrows + i);
32 Tpetra::global_size_t dummy =
33 Teuchos::OrdinalTraits<Tpetra::global_size_t>::invalid();
34 Teuchos::RCP<const map_t> map = rcp(
new map_t(dummy, myRows, 0, comm));
37 JBlock = rcp(
new matrix_t(map, nnz));
38 Teuchos::Array<gno_t> myCols(nnz);
39 Teuchos::Array<zscalar_t> myVals(nnz);
41 for (
size_t i = 0; i < myNrows; i++) {
42 auto gid = map->getGlobalElement(i);
45 if (gid+multiple <= map->getMaxAllGlobalIndex())
46 myCols[cnt++] = gid+multiple;
47 JBlock->insertGlobalValues(gid, myCols(0,cnt), myVals(0, cnt));
49 JBlock->fillComplete();
52 using IST =
typename Kokkos::Details::ArithTraits<zscalar_t>::val_type;
54 Kokkos::Random_XorShift64_Pool<execution_space_t>;
55 pool_type rand_pool(
static_cast<uint64_t
>(me));
57 Kokkos::fill_random(JBlock->getLocalMatrixDevice().values, rand_pool,
58 static_cast<IST
>(1.),
static_cast<IST
>(9999.));
60 Teuchos::FancyOStream foo(Teuchos::rcp(&std::cout,
false));
61 JBlock->describe(foo, Teuchos::VERB_EXTREME);
67 size_t nIndices = std::max(JBlock->getGlobalNumCols(),
68 JBlock->getGlobalNumRows());
69 Teuchos::Array<gno_t> indices(nIndices);
71 Teuchos::RCP<const map_t> vMapCyclic =
72 getCyclicMap(JBlock->getGlobalNumCols(), indices, np-1,
74 Teuchos::RCP<const map_t> wMapCyclic =
75 getCyclicMap(JBlock->getGlobalNumRows(), indices, np-2,
79 RCP<const graph_t> block_graph = JBlock->getCrsGraph();
80 RCP<graph_t> cyclic_graph = rcp(
new graph_t(*block_graph));
81 cyclic_graph->resumeFill();
82 cyclic_graph->fillComplete(vMapCyclic, wMapCyclic);
83 JCyclic = rcp(
new matrix_t(cyclic_graph));
84 JCyclic->resumeFill();
85 TEUCHOS_ASSERT(block_graph->getLocalNumRows() ==
86 cyclic_graph->getLocalNumRows());
88 auto val_s = JBlock->getLocalMatrixHost().values;
89 auto val_d = JCyclic->getLocalMatrixHost().values;
90 TEUCHOS_ASSERT(val_s.extent(0) == val_d.extent(0));
91 Kokkos::deep_copy(val_d, val_s);
93 JCyclic->fillComplete();
94 JCyclic->describe(foo, Teuchos::VERB_EXTREME);
115 const char *testname,
116 Teuchos::ParameterList ¶ms,
123 Teuchos::RCP<matrix_t> J = (useBlock ? JBlock : JCyclic);
124 int me = J->getRowMap()->getComm()->getRank();
126 std::cout <<
"Running " << testname <<
" with "
127 << (useBlock ?
"Block maps" :
"Cyclic maps")
132 colorer.computeColoring(params);
135 if (!colorer.checkColoring()) {
136 std::cout << testname <<
" with "
137 << (useBlock ?
"Block maps" :
"Cyclic maps")
138 <<
" FAILED: invalid coloring returned"
146 const int numColors = colorer.getNumColors();
151 colorer.computeSeedMatrix(V);
160 Teuchos::RCP<matrix_t> Jp = rcp(
new matrix_t(*J, Teuchos::Copy));
161 Jp->setAllToScalar(
static_cast<zscalar_t>(-1.));
163 colorer.reconstructMatrix(W, *Jp);
166 auto J_local_matrix = J->getLocalMatrixDevice();
167 auto Jp_local_matrix = Jp->getLocalMatrixDevice();
168 const size_t num_local_nz = J->getLocalNumEntries();
170 Kokkos::parallel_reduce(
171 "TpetraCrsColorer::testReconstructedMatrix()",
172 Kokkos::RangePolicy<execution_space_t>(0, num_local_nz),
173 KOKKOS_LAMBDA(
const size_t nz,
int &errorcnt) {
174 if (J_local_matrix.values(nz) != Jp_local_matrix.values(nz)) {
175 printf(
"Error in nonzero comparison %zu: %g != %g",
176 nz, J_local_matrix.values(nz), Jp_local_matrix.values(nz));
184 std::cout << testname <<
" FAILED on rank " << me <<
" with "
185 << (useBlock ?
"Block maps" :
"Cyclic maps")