Multi gpu cleanup only if gpu actually did something

olepoeschl · Feb 19, 2024 · 1e3155d · 1e3155d
1 parent d2864dd
commit 1e3155d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/de/nqueensfaf/impl/GpuSolver.java b/src/main/java/de/nqueensfaf/impl/GpuSolver.java
@@ -368,10 +368,10 @@ private void multiGpu(List<Constellation> constellations) {
 	    throw new RuntimeException("could not wait for termination of GpuSolver: " + e.getMessage());
 	}
 
-	for (var gpu : selectedGpus) {
-	    if (gpu.maxNumOfConstellationsPerRun == 0)
+	for (int gpuIdx = 0; gpuIdx < selectedGpus.size(); gpuIdx++) {
+	    if (firstWorkloads.get(gpuIdx).size() == 0)
 		continue;
-	    gpu.releaseBuffers();
+	    selectedGpus.get(gpuIdx).releaseBuffers();
 	}
     }