saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-23 22:13:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix Vitis-AI EP for memory info into IAllocator move (#4404)

Browse source
This commit is contained in:
jornt-xilinx 2020-07-05 00:00:26 +01:00 committed by GitHub
parent 8bcdefc9c1
commit 0d4a65eede
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc
View file

@ -24,7 +24,7 @@ using namespace ::onnxruntime::logging;
namespace onnxruntime {
constexpr const char* PREFIX = "VITISAI";
constexpr const char* VITISAI = "VITISAI";
typedef std::shared_ptr<pyxir::graph::XGraph> XGraphHolder;
typedef std::shared_ptr<pyxir::graph::XLayer> XLayerHolder;
@ -33,14 +33,11 @@ VitisAIExecutionProvider::VitisAIExecutionProvider(const VitisAIExecutionProvide
: IExecutionProvider{onnxruntime::kVitisAIExecutionProvider}, backend_type_(info.backend_type),
device_id_(info.device_id) {
auto default_allocator_factory = [](int) {
auto memory_info = onnxruntime::make_unique<OrtMemoryInfo>(PREFIX, OrtAllocatorType::OrtDeviceAllocator);
return onnxruntime::make_unique<CPUAllocator>(std::move(memory_info));
};
DeviceAllocatorRegistrationInfo default_memory_info{
OrtMemTypeDefault,
std::move(default_allocator_factory),
[](int) {
return onnxruntime::make_unique<CPUAllocator>(OrtMemoryInfo(VITISAI, OrtAllocatorType::OrtDeviceAllocator));
},
std::numeric_limits<size_t>::max()
};