diff --git a/onnxruntime/core/session/onnxruntime_c_api.cc b/onnxruntime/core/session/onnxruntime_c_api.cc index 36f7f1f60c36e..21d09df5cc4db 100644 --- a/onnxruntime/core/session/onnxruntime_c_api.cc +++ b/onnxruntime/core/session/onnxruntime_c_api.cc @@ -3671,7 +3671,7 @@ OrtStatus* GetInputOutputMemoryInfo(const OrtSession* ort_session, InlinedVector mem_info; ORT_API_RETURN_IF_STATUS_NOT_OK( - session->GetInputOutputMemoryInfo(InferenceSession::SessionInputOutputType::kInput, mem_info)); + session->GetInputOutputMemoryInfo(type, mem_info)); auto num_found = mem_info.size(); if (num_found > num_values) { diff --git a/onnxruntime/test/providers/nv_tensorrt_rtx/test_nv_trt_rtx_ep_util.cc b/onnxruntime/test/providers/nv_tensorrt_rtx/test_nv_trt_rtx_ep_util.cc index 17182ab032f7a..7f7894abdf3d5 100644 --- a/onnxruntime/test/providers/nv_tensorrt_rtx/test_nv_trt_rtx_ep_util.cc +++ b/onnxruntime/test/providers/nv_tensorrt_rtx/test_nv_trt_rtx_ep_util.cc @@ -433,7 +433,7 @@ Ort::IoBinding generate_io_binding( } const OrtMemoryInfo* info; Ort::ThrowOnError(Ort::GetApi().AllocatorGetInfo(allocator, &info)); - Ort::MemoryInfo mem_info(info->name, info->alloc_type, info->device.Id(), info->mem_type); + Ort::MemoryInfo mem_info(info->name.c_str(), info->alloc_type, static_cast(info->device.Id()), info->mem_type); for (int input_idx = 0; input_idx < int(session.GetInputCount()); ++input_idx) { auto input_name = session.GetInputNameAllocated(input_idx, Ort::AllocatorWithDefaultOptions());