free memory after query device

justanhduc · Nov 16, 2020 · 820035e · 820035e
1 parent c554531
commit 820035e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/client.c b/client.c
@@ -151,6 +151,7 @@ int c_wait_server_commands()
                         strcat(tmp, ",");
                 }
                 putenv(tmp);
+                free(freeList);
             }
 
             run_job(&result);

diff --git a/gpu.c b/gpu.c
@@ -22,6 +22,7 @@ int * getFreeGpuList(int *numFree) {
             count++;
             j++;
         }
+        cudaDeviceReset();
     }
     *numFree = count;
     return gpuList;

diff --git a/jobs.c b/jobs.c
@@ -637,7 +637,8 @@ int next_run_job()
             if (p->gpus) {
                 int numFree;
                 /* get number of free GPUs at the moment */
-                getFreeGpuList(&numFree);
+                int * list = getFreeGpuList(&numFree);
+                free(list);
 
                 /* GPU mem takes some time to be allocated
                  * if there are many processes in queue,