TestHelper.h File Reference

#include <cassert>
#include <cstdio>
#include <string.h>

Go to the source code of this file.

Data Structures
class	Argv< N >

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

Function Documentation

entry()

int entry	(	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 155 of file Driver.cpp.

{
  // Usage:
  //  [Command] --frontend [Frontend .so path] --frontend-arg ...
  std::unique_ptr<FrontendZone> frontend_zone;
 
  // Simple argument parser (based on map)
  std::map<std::string, std::function<void(const std::string &arg)>> argparse;
 
  argparse["--frontend"] = [&](const std::string &path) {
    frontend_zone = std::make_unique<FrontendZone>(path);
  };
 
  argparse["--frontend-arg"] = [&](const std::string &arg) { frontend_zone->append(arg); };
 
  if (argc < 2)
  {
    std::cerr << "Usage:" << std::endl;
    std::cerr << "[Command] --frontend [.so path]" << std::endl;
    std::cerr << "          --frontend-arg [argument] ..." << std::endl;
    return 255;
  }
 
  for (int n = 1; n < argc; n += 2)
  {
    const std::string tag{argv[n]};
    const std::string arg{argv[n + 1]};
 
    auto it = argparse.find(tag);
 
    if (it == argparse.end())
    {
      std::cerr << "Option '" << tag << "' is not supported" << std::endl;
      return 255;
    }
 
    it->second(arg);
  }
 
  assert(frontend_zone != nullptr);
 
  auto frontend = frontend_zone->factory()->make(frontend_zone->args());
 
  auto bundle = frontend->load();
 
  // dump
  dump(bundle.module());
 
  // todo : dump data
 
  return 0;
}