onnxruntime/docs/how-to/add-execution-provider.md
Faith Xu 0f6f0d9bbb
Major updates/restructure for documentation (#7609)
* Update documentation.

Updates documentation.

* Update tf-get-started.md
2021-05-10 15:19:37 -07:00

2.4 KiB

title parent nav_order
Add a new execution provider How to 7

Add a new Execution Provider to ONNX Runtime

{: .no_toc }

Contents

{: .no_toc }

  • TOC placeholder {:toc}

Create the Execution Provider

  1. Create a folder under onnxruntime/core/providers
  2. Create a folder under include/onnxruntime/core/providers, it should has the same name as the first step.
  3. Create a new class, which must inherit from IExecutionProvider. The source code should be put in 'onnxruntime/core/providers/[your_provider_name]'
  4. Create a new header file under include/onnxruntime/core/providers/[your_provider_name]. The file should provide one function for creating an OrtProviderFactoryInterface. You may use 'include/onnxruntime/core/providers/cpu/cpu_provider_factory.h' as a template. You don't need to provide a function for creating MemoryInfo.
  5. Put a symbols.txt under 'onnxruntime/core/providers/[your_provider_name]'. The file should contain all the function names that would be exported from you provider. Usually, just a single function for creating provider factory is enough.
  6. Add your provider in onnxruntime_providers.cmake. Build it as a static lib.
  7. Add one line in cmake/onnxruntime.cmake, to the 'target_link_libraries' function call. Put your provider there.

Examples:

Use the Execution Provider

  1. Create a factory for that provider, by using the c function you exported in 'symbols.txt'
  2. Put the provider factory into session options
  3. Create session from that session option

Example:

  OrtEnv* env;
  OrtInitialize(ORT_LOGGING_LEVEL_WARNING, "test", &env)
  OrtSessionOptions* session_option = OrtCreateSessionOptions();
  OrtProviderFactoryInterface** factory;
  OrtCreateCUDAExecutionProviderFactory(0, &factory);
  OrtSessionOptionsAppendExecutionProvider(session_option, factory);
  OrtReleaseObject(factory);
  OrtCreateSession(env, model_path, session_option, &session);