nnkit Namespace Reference

Namespaces
namespace	support

Data Structures
struct	Action
struct	Backend
class	BackendPlugin
struct	CmdlineArguments
struct	TensorContext
class	VectorArguments

Functions
std::unique_ptr< BackendPlugin >	make_backend_plugin (const std::string &path)

Function Documentation

make_backend_plugin()

std::unique_ptr< BackendPlugin > nnkit::make_backend_plugin

(

const std::string &

path

)

Definition at line 52 of file BackendPlugin.cpp.

{
  if (path.empty())
  {
    throw std::runtime_error{"Backend library does not defined"};
  }
 
  void *handle;
  BackendPlugin::Entry entry;
 
  // NOTE Some backend (such as tflite) needs multithreading support (std::thread).
  //
  //      std::thread in libstdc++.so includes weak symbols for pthread_XXX functions,
  //      and these weak symbols should be overridden by strong symbols in libpthread.so.
  //      If not, std::thread will not work correctly.
  //
  //      RTLD_GLOBAL flag is necessary to allow weak symbols to be overridden.
  handle = dlopen(path.c_str(), RTLD_LAZY | RTLD_GLOBAL);
  if (handle == nullptr)
  {
    std::cerr << dlerror() << std::endl;
    exit(1);
  }
 
  char *error;
  entry = reinterpret_cast<BackendPlugin::Entry>(dlsym(handle, "make_backend"));
  if ((error = dlerror()) != nullptr)
  {
    dlclose(handle);
    std::cerr << error << std::endl;
    exit(1);
  }
 
  return std::make_unique<BackendPlugin>(handle, entry);
}

References entry().

Referenced by main().