astra-sim-alibabacloud/astra-sim/system/CollectivePhase.cc

/******************************************************************************
This source code is licensed under the MIT license found in the
LICENSE file in the root directory of this source tree.
*******************************************************************************/

#include "CollectivePhase.hh"
#include "astra-sim/system/collective/Algorithm.hh"
namespace AstraSim {
CollectivePhase::CollectivePhase(
    Sys* generator,
    int queue_id,
    Algorithm* algorithm) {
  this->generator = generator;
  this->queue_id = queue_id;
  this->algorithm = algorithm;
  this->enabled = true;
  this->initial_data_size = algorithm->data_size;
  this->final_data_size = algorithm->final_data_size;
  this->comm_type = algorithm->comType;
  this->enabled = algorithm->enabled;
}
CollectivePhase::CollectivePhase() {
  queue_id = -1;
  generator = nullptr;
  algorithm = nullptr;
}
void CollectivePhase::init(BaseStream* stream) {
  if (algorithm != nullptr) {
    algorithm->init(stream);
  }
}
} // namespace AstraSim