diff --git a/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_gpu_agent.cpp b/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_gpu_agent.cpp index 78c114a929..7ac0c0842c 100644 --- a/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_gpu_agent.cpp +++ b/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_gpu_agent.cpp @@ -926,7 +926,9 @@ hsa_status_t GpuAgent::DmaCopyStatus(core::Agent& dst_agent, core::Agent& src_ag } if (!!!blits_[BlitDevToHost]->PendingBytes()) { - *engine_ids_mask |= HSA_AMD_SDMA_ENGINE_1; + *engine_ids_mask |= properties_.NumSdmaEngines > 1 ? + HSA_AMD_SDMA_ENGINE_1 : + HSA_AMD_SDMA_ENGINE_0; } // Find a free xGMI SDMA engine for H2D/D2H though it may be lower bandwidth for (int i = 0; i < properties_.NumSdmaXgmiEngines; i++) {