diff --git a/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc b/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc index 699f0cb2e6..59e59fa21e 100644 --- a/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc +++ b/onnxruntime/core/providers/vitisai/vitisai_execution_provider.cc @@ -24,7 +24,7 @@ using namespace ::onnxruntime::logging; namespace onnxruntime { -constexpr const char* PREFIX = "VITISAI"; +constexpr const char* VITISAI = "VITISAI"; typedef std::shared_ptr XGraphHolder; typedef std::shared_ptr 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(PREFIX, OrtAllocatorType::OrtDeviceAllocator); - return onnxruntime::make_unique(std::move(memory_info)); - }; - DeviceAllocatorRegistrationInfo default_memory_info{ OrtMemTypeDefault, - std::move(default_allocator_factory), + [](int) { + return onnxruntime::make_unique(OrtMemoryInfo(VITISAI, OrtAllocatorType::OrtDeviceAllocator)); + }, std::numeric_limits::max() };