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

Update calls to OrtArenaCfg constructor to pass additional parameter. (#16104)

Browse source 
Update calls to OrtArenaCfg constructor to pass additional parameter.

Updating some call sites after change in #15983. Fix CI build.
This commit is contained in:
Edward Chen 2023-05-25 21:41:42 -07:00 committed by GitHub
parent 2cf0ae7d01
commit 4bfb8d3303
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/core/providers/cann/cann_execution_provider.cc
View file

@ -1455,6 +1455,7 @@ void CANNExecutionProvider::RegisterAllocator(AllocatorManager& allocator_manage
static_cast<int>(info_.arena_extend_strategy),
-1,
-1,
-1,
-1)},
true,
false);

7
onnxruntime/core/providers/rocm/rocm_execution_provider.cc
View file

@ -120,7 +120,12 @@ AllocatorPtr ROCMExecutionProvider::CreateRocmAllocator(OrtDevice::DeviceId devi
device_id,
true,
{default_memory_arena_cfg ? *default_memory_arena_cfg
: OrtArenaCfg(gpu_mem_limit, static_cast<int>(arena_extend_strategy), -1, -1, -1)},
: OrtArenaCfg(gpu_mem_limit,
static_cast<int>(arena_extend_strategy),
-1,
-1,
-1,
-1)},
// make it stream aware
true,
// enable cross stream sharing?