Merge pull request #24 from Project-HAMi/fix_utilization

fix error code on device utilization
Project-HAMi · Oct 15, 2024 · a9ab3b1 · a9ab3b1
2 parents 6b2aed4 + f756e3c
commit a9ab3b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/multiprocess/multiprocess_utilization_watcher.c b/src/multiprocess/multiprocess_utilization_watcher.c
@@ -150,7 +150,7 @@ int get_used_gpu_utilization(int *userutil,int *sysprocnum) {
       res = nvmlDeviceGetProcessUtilization(device,processes_sample,&processes_num,microsec);
       LOG_DEBUG("processes_num=%d\n",processes_num);
       LOG_DEBUG("Device UUID: %s\n", uuid);
-      if (res==NVML_ERROR_INSUFFICIENT_SIZE){
+      if ((res==NVML_ERROR_INSUFFICIENT_SIZE) || (res==NVML_ERROR_NOT_FOUND)){
         userutil[cudadev] = 0;
         for (i=0; i<infcount; i++){
           proc = find_proc_by_hostpid(infos[i].pid);