From b6134311e04deb3d172740a946f676e650861d1d Mon Sep 17 00:00:00 2001 From: chen qingxiang <67679556+godchen0212@users.noreply.github.com> Date: Fri, 25 Sep 2020 14:25:50 +0800 Subject: [PATCH] fix gpu check bug (#3866) * fix gpu check bug Signed-off-by: godchen0212 * format code Signed-off-by: godchen0212 Signed-off-by: shengjun.li --- core/src/server/init/GpuChecker.cpp | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/core/src/server/init/GpuChecker.cpp b/core/src/server/init/GpuChecker.cpp index 4d3dba8850..dbd19d124a 100644 --- a/core/src/server/init/GpuChecker.cpp +++ b/core/src/server/init/GpuChecker.cpp @@ -188,11 +188,7 @@ GpuChecker::CheckGpuEnvironment() { char device_name[NVML_DEVICE_NAME_BUFFER_SIZE]; int major, minor; - for (uint32_t i = 0; i < device_count; i++) { - if (gpu_sets.find(i) == gpu_sets.end()) { - continue; - } - + for (uint32_t i = 0; i < gpu_sets.size(); i++) { nvmlDevice_t device; nvmlresult = nvmlDeviceGetHandleByIndex(i, &device); fiu_do_on("GpuChecker.CheckGpuEnvironment.nvml_get_device_handle_fail", nvmlresult = NVML_ERROR_UNKNOWN);