From e6f66f660cf2ad8bb25fc0e0d0188e664cb9b241 Mon Sep 17 00:00:00 2001 From: Weixing Zhang Date: Fri, 30 Apr 2021 09:53:05 -0700 Subject: [PATCH] missed change for external allocator in ROCm EP. (#7505) --- .../core/providers/rocm/rocm_execution_provider_info.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/onnxruntime/core/providers/rocm/rocm_execution_provider_info.cc b/onnxruntime/core/providers/rocm/rocm_execution_provider_info.cc index 8d1be8b0c5..ba63e185c7 100644 --- a/onnxruntime/core/providers/rocm/rocm_execution_provider_info.cc +++ b/onnxruntime/core/providers/rocm/rocm_execution_provider_info.cc @@ -57,6 +57,9 @@ ROCMExecutionProviderInfo ROCMExecutionProviderInfo::FromProviderOptions(const P arena_extend_strategy_mapping, info.arena_extend_strategy) .Parse(options)); + ROCMExecutionProviderExternalAllocatorInfo alloc_info{alloc, free}; + info.external_allocator_info = alloc_info; + return info; }