From 4bfb8d3303c72727f9fbfa529abdacd7b2fed0e5 Mon Sep 17 00:00:00 2001
From: Edward Chen <18449977+edgchen1@users.noreply.github.com>
Date: Thu, 25 May 2023 21:41:42 -0700
Subject: [PATCH] Update calls to OrtArenaCfg constructor to pass additional
 parameter. (#16104)

Update calls to OrtArenaCfg constructor to pass additional parameter.

Updating some call sites after change in #15983. Fix CI build.
---
 onnxruntime/core/providers/cann/cann_execution_provider.cc | 1 +
 onnxruntime/core/providers/rocm/rocm_execution_provider.cc | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

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<int>(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<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?