From 16246764ce4fcd70b150ead6d313f454c276f420 Mon Sep 17 00:00:00 2001 From: Felix Kuehling Date: Thu, 5 Jul 2018 15:32:14 -0400 Subject: [PATCH] Fix wrong loop termination condition Compare with gpu_mem_count instead of deprecated NUM_OF_SUPPORTED_GPUS to prevent overflows in case no dGPUs are present. Change-Id: I71fcb7503ba4c20bffadbdb04cefc4e4027a7df7 Signed-off-by: Felix Kuehling [ROCm/ROCR-Runtime commit: d3228f363e458e93bf3b5823206814275aaeeff4] --- projects/rocr-runtime/src/fmm.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/rocr-runtime/src/fmm.c b/projects/rocr-runtime/src/fmm.c index 5de949ec82..7b9ccd37e4 100644 --- a/projects/rocr-runtime/src/fmm.c +++ b/projects/rocr-runtime/src/fmm.c @@ -223,11 +223,11 @@ static void print_device_id_array(uint32_t *device_id_array, uint32_t device_id_ static int32_t find_first_dgpu(HSAuint32 *gpu_id) { - int32_t i; + uint32_t i; *gpu_id = NON_VALID_GPU_ID; - for (i = 0; i < NUM_OF_SUPPORTED_GPUS; i++) { + for (i = 0; i < gpu_mem_count; i++) { if (gpu_mem[i].gpu_id == NON_VALID_GPU_ID) continue; if (!topology_is_dgpu(gpu_mem[i].device_id))