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

missed change for external allocator in ROCm EP. (#7505)

Browse source
This commit is contained in:
Weixing Zhang 2021-04-30 09:53:05 -07:00 committed by GitHub
parent 70e67ddd2b
commit e6f66f660c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/core/providers/rocm/rocm_execution_provider_info.cc
View file

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