saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix brainslice break

Browse source
This commit is contained in:
souptc 2019-02-01 15:54:22 -08:00 committed by Changming Sun
parent 3b061d60a9
commit 214c1b88e3
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/test/perftest/testenv.cc
View file

@ -55,7 +55,7 @@ Status SessionFactory::create(std::shared_ptr<::onnxruntime::InferenceSession>&
#endif
} else if (provider == onnxruntime::kBrainSliceExecutionProvider) {
#if USE_BRAINSLICE
RegisterExecutionProvider(sess.get(), onnxruntime::test::DefaultBrainsliceExecutionProvider());
RegisterExecutionProvider(sess.get(), onnxruntime::test::DefaultBrainSliceExecutionProvider());
#else
ORT_THROW("This executable was not built with BrainSlice");
#endif

4
onnxruntime/test/util/default_providers.cc
View file

@ -12,7 +12,7 @@ std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_CPU(in
std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_CUDA(int device_id);
std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_Mkldnn(int use_arena);
std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_Nuphar(int device_id, const char*);
std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_BrainSlice(int id, bool f, const char*, const char*, const char*);
std::shared_ptr<IExecutionProviderFactory> CreateExecutionProviderFactory_BrainSlice(uint32_t id, bool f, const char*, const char*, const char*);
namespace test {
@ -47,7 +47,7 @@ std::unique_ptr<IExecutionProvider> DefaultNupharExecutionProvider() {
std::unique_ptr<IExecutionProvider> DefaultBrainSliceExecutionProvider() {
#ifdef USE_BRAINSLICE
return CreateExecutionProviderFactory_BrainSlice(0, true, "testdata/firmwares/onnx_rnns/instructions.bin", "testdata/firmwares/onnx_rnns/data.bin", "testdata/firmwares/onnx_rnns/schema.bin", &f));
return CreateExecutionProviderFactory_BrainSlice(0, true, "testdata/firmwares/onnx_rnns/instructions.bin", "testdata/firmwares/onnx_rnns/data.bin", "testdata/firmwares/onnx_rnns/schema.bin")->CreateProvider();
#else
return nullptr;
#endif