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);