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