From b222a8e01bdf7537c3ac633fa23e45e62150f14d Mon Sep 17 00:00:00 2001 From: PeixuanZuo <94887879+PeixuanZuo@users.noreply.github.com> Date: Fri, 6 Jan 2023 15:51:25 +0800 Subject: [PATCH] [Fix] build error with MIGraphX tag rocm-5.4.0 (#14141) ### Description Fix the error https://github.com/microsoft/onnxruntime/issues/14126 ### Motivation and Context Co-authored-by: peixuanzuo --- .../core/providers/migraphx/migraphx_execution_provider.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/migraphx/migraphx_execution_provider.cc b/onnxruntime/core/providers/migraphx/migraphx_execution_provider.cc index fb64d25dc3..8a4c7f46bd 100644 --- a/onnxruntime/core/providers/migraphx/migraphx_execution_provider.cc +++ b/onnxruntime/core/providers/migraphx/migraphx_execution_provider.cc @@ -1044,7 +1044,7 @@ Status MIGraphXExecutionProvider::Compile(const std::vector& delete static_cast(state); }; - compute_info.compute_func = [this](FunctionState state, const OrtApi* /* api */, OrtKernelContext* context) { + compute_info.compute_func = [this](FunctionState state, const OrtApi* api, OrtKernelContext* context) { Ort::KernelContext ctx(context); MIGraphXFuncState* mgx_state = reinterpret_cast(state); @@ -1173,7 +1173,7 @@ Status MIGraphXExecutionProvider::Compile(const std::vector& #ifdef MIGRAPHX_STREAM_SYNC void* rocm_stream; - api->KernelContext_GetGPUComputeStream(context, &rocm_stream); + Ort::ThrowOnError(api->KernelContext_GetGPUComputeStream(context, &rocm_stream)); auto prog_outputs = prog.run_async(m, static_cast(rocm_stream)); #else auto prog_outputs = prog.eval(m);