Method evalDiff

compiler/circle-eval-diff/src/CircleEvalDiff.cpp:201–246

}

void CircleEvalDiff::evalDiff(void) const
{
  auto first_input_loader = circle_eval_diff::makeDataLoader(
    _ctx->first_input_data_path, _ctx->input_format, ::inputs_of(_first_module.get()));
  auto second_input_loader = circle_eval_diff::makeDataLoader(
    _ctx->second_input_data_path, _ctx->input_format, ::inputs_of(_second_module.get()));

  for (uint32_t data_idx = 0; data_idx < first_input_loader->size(); data_idx++)
  {
    std::cout << "Evaluating " << data_idx << "'th data" << std::endl;

    auto first_data = first_input_loader->get(data_idx);
    auto second_data = second_input_loader->get(data_idx);

    auto first_output = interpret(_first_module.get(), first_data);
    auto second_output = interpret(_second_module.get(), second_data);

    for (auto &metric : _metrics)
    {
      metric->accumulate(first_output, second_output);
    }

    if (_ctx.get()->output_prefix.empty())
      continue;

    for (uint32_t i = 0; i < first_output.size(); i++)
    {
      auto out = first_output[i];
      writeDataToFile(_ctx.get()->output_prefix + "." + std::to_string(data_idx) + ".first.output" +
                        std::to_string(i),
                      (char *)(out->buffer()), out->byte_size());
    }
    for (uint32_t i = 0; i < second_output.size(); i++)
    {
      auto out = second_output[i];
      writeDataToFile(_ctx.get()->output_prefix + "." + std::to_string(data_idx) +
                        ".second.output" + std::to_string(i),
                      (char *)(out->buffer()), out->byte_size());
    }
  }

  for (auto &metric : _metrics)
  {
    std::cout << metric.get() << std::endl;
  }
}

} // namespace circle_eval_diff