Skip to content

Commit

Permalink
correct code for cases when number of processors is less than CTF_PPN
Browse files Browse the repository at this point in the history
  • Loading branch information
solomonik committed Nov 9, 2023
1 parent 28110cf commit 24b9ff8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/contraction/contraction.cxx
Expand Up @@ -4441,7 +4441,7 @@ namespace CTF_int {
if (C->wrld->ppn != 1 && !is_sparse()){
TAU_FSTART(node_aware_remapping_calc);
std::vector<int> pe_grid(orig_topo.lens, orig_topo.lens + orig_topo.order);
std::vector<std::vector<int> > inter_node_grids = CTF_int::get_inter_node_grids(pe_grid, C->wrld->np/C->wrld->ppn);
std::vector<std::vector<int> > inter_node_grids = CTF_int::get_inter_node_grids(pe_grid, std::max(1,C->wrld->np/C->wrld->ppn));
//std::vector< std::vector<int> > intra_node_grids = CTF_int::get_all_shapes(C->wrld->ppn()){
int * intra_node_lens = (int*)CTF_int::alloc((orig_topo.order)*sizeof(int));
int64_t best_topo_index;
Expand Down

0 comments on commit 24b9ff8

Please sign in to comment.