pytorch

1
#include <fstream>
2
#include <iterator>
3
#include <string>
4

5
#include "binaries/benchmark_args.h"
6
#include "binaries/benchmark_helper.h"
7

8

9
int main(int argc, char** argv) {
10
  caffe2::GlobalInit(&argc, &argv);
11
  benchmark(
12
      argc,
13
      argv,
14
      FLAGS_backend,
15
      FLAGS_init_net,
16
      FLAGS_input,
17
      FLAGS_input_dims,
18
      FLAGS_input_file,
19
      FLAGS_input_type,
20
      FLAGS_iter,
21
      FLAGS_measure_memory,
22
      FLAGS_net,
23
      FLAGS_output,
24
      FLAGS_output_folder,
25
      FLAGS_run_individual,
26
      FLAGS_sleep_before_run,
27
      FLAGS_sleep_between_iteration,
28
      FLAGS_sleep_between_net_and_operator,
29
      FLAGS_text_output,
30
      FLAGS_warmup,
31
      FLAGS_wipe_cache);
32
}
33