fixed gpu list

justanhduc · Nov 13, 2020 · 9ea170e · 9ea170e
1 parent dde0298
commit 9ea170e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gpu.c b/gpu.c
@@ -8,7 +8,7 @@
 int * getFreeGpuList(int *numFree) {
     int * gpuList;
     int nDevices;
-    int i, count = 0;
+    int i, j = 0, count = 0;
 
     cudaGetDeviceCount(&nDevices);
     gpuList = (int *) malloc(nDevices * sizeof(int));
@@ -18,11 +18,11 @@ int * getFreeGpuList(int *numFree) {
         size_t totalMem;
         cudaMemGetInfo(&freeMem, &totalMem);
         if (freeMem > .9 * totalMem) {
-            gpuList[i] = i;
+            gpuList[j] = i;
             count++;
+            j++;
         }
     }
     *numFree = count;
-    free(gpuList);
     return gpuList;
 }