diff --git a/onnxruntime/python/onnxruntime_pybind_state.cc b/onnxruntime/python/onnxruntime_pybind_state.cc index aa4fce0425..8570901f52 100644 --- a/onnxruntime/python/onnxruntime_pybind_state.cc +++ b/onnxruntime/python/onnxruntime_pybind_state.cc @@ -58,6 +58,9 @@ #ifdef USE_NUPHAR #include "core/providers/nuphar/nuphar_provider_factory.h" #endif +#ifdef USE_BRAINSLICE +#include "core/providers/brainslice/brainslice_provider_factory.h" +#endif namespace onnxruntime { std::shared_ptr CreateExecutionProviderFactory_CPU(int use_arena); @@ -65,7 +68,7 @@ std::shared_ptr CreateExecutionProviderFactory_CUDA(i std::shared_ptr CreateExecutionProviderFactory_Tensorrt(); std::shared_ptr CreateExecutionProviderFactory_Mkldnn(int use_arena); std::shared_ptr CreateExecutionProviderFactory_Nuphar(int device_id, const char*); -std::shared_ptr CreateExecutionProviderFactory_BrainSlice(int ip, int, int, bool, const char*, const char*, const char*); +std::shared_ptr CreateExecutionProviderFactory_BrainSlice(uint32_t ip, int, int, bool, const char*, const char*, const char*); } // namespace onnxruntime #if defined(_MSC_VER) @@ -215,12 +218,19 @@ void InitializeSession(InferenceSession* sess) { RegisterExecutionProvider(sess, *onnxruntime::CreateExecutionProviderFactory_Mkldnn(enable_cpu_mem_arena ? 1 : 0)); } #endif + #if 0 //USE_NUPHAR { RegisterExecutionProvider(sess, *onnxruntime::CreateExecutionProviderFactory_Nuphar(0, "")); } #endif +#ifdef USE_BRAINSLICE + { + RegisterExecutionProvider(sess, *onnxruntime::CreateExecutionProviderFactory_BrainSlice(0, -1, -1, false, "", "", "")); + } +#endif + status = sess->Initialize(); if (!status.IsOK()) { throw std::runtime_error(status.ToString().c_str());