little bugfix

[trigger-nightly-build]
olepoeschl · Feb 19, 2024 · c8fef51 · c8fef51
1 parent cadb6bb
commit c8fef51
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/de/nqueensfaf/impl/GpuSolver.java b/src/main/java/de/nqueensfaf/impl/GpuSolver.java
@@ -294,15 +294,16 @@ private void multiGpu(List<Constellation> constellations) {
 		toIndex = firstWorkloadSize;
 
 	    var gpuFirstWork = constellations.subList(fromIndex, toIndex);
-	    if (gpuFirstWork.size() == 0)
+	    if (gpuFirstWork.size() == 0) {
+		firstWorkloads.add(new ArrayList<Constellation>());
 		continue;
+	    }
 
 	    var gpuFirstWorkload = fillWithPseudoConstellations(gpuFirstWork, selectedGpus.get(gpuIdx).workgroupSize);
 	    firstWorkloads.add(gpuFirstWorkload);
 
 	    var gpu = selectedGpus.get(gpuIdx);
 	    gpu.createBuffers(gpuFirstWorkload.size());
-	    gpu.maxNumOfConstellationsPerRun = gpuFirstWorkload.size();
 
 	    fromIndex = toIndex;
 	}
@@ -314,7 +315,7 @@ private void multiGpu(List<Constellation> constellations) {
 	    final int gpuIdx = idx;
 	    final var gpu = selectedGpus.get(idx);
 
-	    if (gpu.maxNumOfConstellationsPerRun == 0)
+	    if (firstWorkloads.get(gpuIdx).size() == 0)
 		continue;
 
 	    executor.execute(() -> {