From 0d4a65eede737b10cbbdce563b413199b29c06c2 Mon Sep 17 00:00:00 2001
From: jornt-xilinx <65705538+jornt-xilinx@users.noreply.github.com>
Date: Sun, 5 Jul 2020 00:00:26 +0100
Subject: [PATCH] Fix Vitis-AI EP for memory info into IAllocator move (#4404)

---
 .../providers/vitisai/vitisai_execution_provider.cc   | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

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<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()
   };