#include <iostream>
#include "common.h"
#include "dnnmark.h"
#include "usage.h"

using namespace dnnmark;

int main(int argc, char **argv) {
  INIT_FLAGS(argc, argv);
  INIT_LOG(argv);
  LOG(INFO) << "DNNMark suites: Start...";
  DNNMark<TestType> dnnmark(58, FLAGS_mmap);
  dnnmark.ParseAllConfig(FLAGS_config);

  dnnmark.Initialize();
  // Warm up
  if (FLAGS_warmup) {
    for (int i = 0; i < 5; i++) {
      dnnmark.Forward();
      dnnmark.Backward();
    }
  }
  dnnmark.GetTimer()->Clear();

  // Real benchmark
  for (int i = 0; i < FLAGS_iterations; i++) {
    LOG(INFO) << "Iteration " << i;
    dnnmark.Forward();
    dnnmark.Backward();
  }
  dnnmark.GetTimer()->SumRecords();

  dnnmark.TearDown();

  LOG(INFO) << "Total running time(ms): " << dnnmark.GetTimer()->GetTotalTime();
  LOG(INFO) << "DNNMark suites: Tear down...";
  printf("Total running time(ms): %f\n", dnnmark.GetTimer()->GetTotalTime());
  return 0;
}