Driver.cpp File Reference

#include "H5Writer.h"
#include <arser/arser.h>

Go to the source code of this file.

Functions
int	entry (const int argc, char **argv)
	Dump IR for given arguments.

Function Documentation

entry()

int entry	(	const int	argc,
		char **	argv
	)

Dump IR for given arguments.

Call example: $ ./build/compiler/encodump/encodump \ –frontend build/compiler/enco/frontend/caffe/libenco_caffe_frontend.so \ –frontend-arg build/compiler/enco/test/caffe/Convolution_003.prototxt \ –frontend-arg build/compiler/enco/test/caffe/Convolution_003.caffemodel

HDF5 layout is like below

GROUP "/" ㄴGROUP "name" ㄴATTRIBUTE "0" ㄴDATA (0): "input_01:0" ㄴATTRIBUTE "1" ㄴDATA (0): "input_02:0" ㄴGROUP "value" ㄴDATASET "0" ㄴDATA ... ㄴDATASET "1" ㄴDATA ...

Definition at line 23 of file Driver.cpp.

{
  arser::Arser arser("Generate min/max data to test minmax-embedder");
  arser.add_argument("--num_inputs")
    .type(arser::DataType::INT32)
    .default_value(1)
    .help("number of input layers (default:1)");
  arser.add_argument("--num_ops")
    .type(arser::DataType::INT32)
    .default_value(1)
    .help("number of operators (default:1)");
  arser.add_argument("minmax").help("path to generated minmax data");
 
  try
  {
    arser.parse(argc, argv);
  }
  catch (const std::runtime_error &err)
  {
    std::cout << err.what() << std::endl;
    std::cout << arser;
    return 255;
  }
 
  auto num_inputs = arser.get<int>("--num_inputs");
  auto num_ops = arser.get<int>("--num_ops");
  auto data_output_path = arser.get<std::string>("minmax");
 
  ModelSpec mspec;
  {
    if (num_inputs <= 0 || num_ops <= 0)
    {
      std::cout << "num_inputs and num_ops must be positive integers." << std::endl;
      return 255;
    }
    mspec.n_inputs = num_inputs;
    mspec.n_ops = num_ops;
  }
  H5Writer writer(mspec, data_output_path);
  writer.dump();
 
  return EXIT_SUCCESS;
}

References minmax_embedder_test::H5Writer::dump(), and minmax_embedder_test::ModelSpec::n_inputs.