sweepscheduler_constrdestr.cc

Go to the documentation of this file.

#include "sweepscheduler.h"
 
#include "chi_runtime.h"
#include "chi_log.h"
 
// ###################################################################
/**Sweep scheduler constructor*/
chi_mesh::sweep_management::SweepScheduler::SweepScheduler(
  SchedulingAlgorithm in_scheduler_type,
  chi_mesh::sweep_management::AngleAggregation& in_angle_agg,
  SweepChunk& in_sweep_chunk)
  : scheduler_type_(in_scheduler_type),
    angle_agg_(in_angle_agg),
    sweep_chunk_(in_sweep_chunk),
    sweep_event_tag_(Chi::log.GetRepeatingEventTag("Sweep Timing")),
    sweep_timing_events_tag_(
      {Chi::log.GetRepeatingEventTag("Sweep Chunk Only Timing"),
       sweep_event_tag_})
 
{
  angle_agg_.InitializeReflectingBCs();
 
  if (scheduler_type_ == SchedulingAlgorithm::DEPTH_OF_GRAPH)
    InitializeAlgoDOG();
 
  //=================================== Initialize delayed upstream data
  for (auto& angsetgrp : in_angle_agg.angle_set_groups)
    for (auto& angset : angsetgrp.AngleSets())
      angset->InitializeDelayedUpstreamData();
 
  //=================================== Get local max num messages accross
  //                                    anglesets
  int local_max_num_messages = 0;
  for (auto& angsetgrp : in_angle_agg.angle_set_groups)
    for (auto& angset : angsetgrp.AngleSets())
      local_max_num_messages =
        std::max(angset->GetMaxBufferMessages(), local_max_num_messages);
 
  //=================================== Reconcile all local maximums
  int global_max_num_messages = 0;
  MPI_Allreduce(&local_max_num_messages,
                &global_max_num_messages,
                1,
                MPI_INT,
                MPI_MAX,
                Chi::mpi.comm);
 
  //=================================== Propogate items back to sweep buffers
  for (auto& angsetgrp : in_angle_agg.angle_set_groups)
    for (auto& angset : angsetgrp.AngleSets())
      angset->SetMaxBufferMessages(global_max_num_messages);
}
 
// ###################################################################
/**Returns the referenced sweep chunk.*/
chi_mesh::sweep_management::SweepChunk&
chi_mesh::sweep_management::SweepScheduler::GetSweepChunk()
{
  return sweep_chunk_;
}