diff --git a/onnxruntime/core/providers/cann/cann_execution_provider.cc b/onnxruntime/core/providers/cann/cann_execution_provider.cc index 6db63752c5..e0a2db0626 100644 --- a/onnxruntime/core/providers/cann/cann_execution_provider.cc +++ b/onnxruntime/core/providers/cann/cann_execution_provider.cc @@ -1455,6 +1455,7 @@ void CANNExecutionProvider::RegisterAllocator(AllocatorManager& allocator_manage static_cast(info_.arena_extend_strategy), -1, -1, + -1, -1)}, true, false); diff --git a/onnxruntime/core/providers/rocm/rocm_execution_provider.cc b/onnxruntime/core/providers/rocm/rocm_execution_provider.cc index 14bad1f3fe..9844eb5669 100644 --- a/onnxruntime/core/providers/rocm/rocm_execution_provider.cc +++ b/onnxruntime/core/providers/rocm/rocm_execution_provider.cc @@ -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(arena_extend_strategy), -1, -1, -1)}, + : OrtArenaCfg(gpu_mem_limit, + static_cast(arena_extend_strategy), + -1, + -1, + -1, + -1)}, // make it stream aware true, // enable cross stream sharing?