From e57b5116d6eb83283aab2e33bf08206b9053dc8f Mon Sep 17 00:00:00 2001 From: Jesse Benson Date: Fri, 8 Feb 2019 10:46:52 -0800 Subject: [PATCH] BrainSlice parameter represents the IP. Update parameter name to match --- onnxruntime/python/onnxruntime_pybind_state.cc | 2 +- onnxruntime/test/util/default_providers.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/python/onnxruntime_pybind_state.cc b/onnxruntime/python/onnxruntime_pybind_state.cc index 91f5716c0b..939bfa4505 100644 --- a/onnxruntime/python/onnxruntime_pybind_state.cc +++ b/onnxruntime/python/onnxruntime_pybind_state.cc @@ -61,7 +61,7 @@ std::shared_ptr CreateExecutionProviderFactory_CPU(in std::shared_ptr CreateExecutionProviderFactory_CUDA(int device_id); std::shared_ptr CreateExecutionProviderFactory_Mkldnn(int use_arena); std::shared_ptr CreateExecutionProviderFactory_Nuphar(int device_id, const char*); -std::shared_ptr CreateExecutionProviderFactory_BrainSlice(int id, bool f, const char*, const char*, const char*); +std::shared_ptr CreateExecutionProviderFactory_BrainSlice(int ip, bool f, const char*, const char*, const char*); } // namespace onnxruntime #if defined(_MSC_VER) diff --git a/onnxruntime/test/util/default_providers.cc b/onnxruntime/test/util/default_providers.cc index 7e754b1f93..bae04b920d 100644 --- a/onnxruntime/test/util/default_providers.cc +++ b/onnxruntime/test/util/default_providers.cc @@ -12,7 +12,7 @@ std::shared_ptr CreateExecutionProviderFactory_CPU(in std::shared_ptr CreateExecutionProviderFactory_CUDA(int device_id); std::shared_ptr CreateExecutionProviderFactory_Mkldnn(int use_arena); std::shared_ptr CreateExecutionProviderFactory_Nuphar(int device_id, const char*); -std::shared_ptr CreateExecutionProviderFactory_BrainSlice(uint32_t id, bool f, const char*, const char*, const char*); +std::shared_ptr CreateExecutionProviderFactory_BrainSlice(uint32_t ip, bool f, const char*, const char*, const char*); std::shared_ptr CreateExecutionProviderFactory_TRT(); namespace test {