Skip to content

Commit

Permalink
POTRF: Add -P and -Q argument
Browse files Browse the repository at this point in the history
Signed-off-by: Joseph Schuchart <[email protected]>
  • Loading branch information
devreal committed Mar 13, 2024
1 parent d2563df commit 910fe0d
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions examples/potrf/testing_dpotrf.cc
Original file line number Diff line number Diff line change
Expand Up @@ -76,14 +76,24 @@ int main(int argc, char **argv)
int P = std::sqrt(world.size());
int Q = (world.size() + P - 1)/P;

if ( (opt = getCmdOption(argv+1, argv + argc, "-P")) != nullptr) {
P = atoi(opt);
}
if ( (opt = getCmdOption(argv+1, argv + argc, "-Q")) != nullptr) {
Q = atoi(opt);
}


if(check && (P>1 || Q>1)) {
std::cerr << "Check is disabled for distributed runs at this time" << std::endl;
check = false;
}

static_assert(ttg::has_split_metadata<MatrixTile<double>>::value);

std::cout << "Creating 2D block cyclic matrix with NB " << NB << " N " << N << " M " << M << " P " << P << std::endl;
if (world.rank() == 0) {
std::cout << "Creating 2D block cyclic matrix with NB " << NB << " N " << N << " M " << M << " P " << P << " Q " << Q << std::endl;
}

parsec_matrix_sym_block_cyclic_t dcA;
parsec_matrix_sym_block_cyclic_init(&dcA, parsec_matrix_type_t::PARSEC_MATRIX_DOUBLE,
Expand Down Expand Up @@ -126,7 +136,6 @@ int main(int argc, char **argv)
auto connected = make_graph_executable(init_tt.get());
assert(connected);
TTGUNUSED(connected);
std::cout << "Graph is connected: " << connected << std::endl;

if (world.rank() == 0) {
std::cout << "==== begin dot ====\n";
Expand Down

0 comments on commit 910fe0d

Please sign in to comment.