mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
1b1bddaab3
Differential Revision: D16469619 Original commit changeset: 670f3549c830 fbshipit-source-id: f55d4cda36f5e29df2df306d33a70158e5a7908b
35 lines
748 B
C++
35 lines
748 B
C++
#include <fstream>
|
|
#include <iterator>
|
|
#include <string>
|
|
|
|
#include "binaries/benchmark_args.h"
|
|
#include "binaries/benchmark_helper.h"
|
|
|
|
using std::make_shared;
|
|
using std::map;
|
|
using std::string;
|
|
using std::vector;
|
|
|
|
int main(int argc, char** argv) {
|
|
caffe2::GlobalInit(&argc, &argv);
|
|
benchmark(
|
|
argc,
|
|
argv,
|
|
FLAGS_backend,
|
|
FLAGS_init_net,
|
|
FLAGS_input,
|
|
FLAGS_input_dims,
|
|
FLAGS_input_file,
|
|
FLAGS_input_type,
|
|
FLAGS_iter,
|
|
FLAGS_net,
|
|
FLAGS_output,
|
|
FLAGS_output_folder,
|
|
FLAGS_run_individual,
|
|
FLAGS_sleep_before_run,
|
|
FLAGS_sleep_between_iteration,
|
|
FLAGS_sleep_between_net_and_operator,
|
|
FLAGS_text_output,
|
|
FLAGS_warmup,
|
|
FLAGS_wipe_cache);
|
|
}
|