Fixed function to find GPU for matrix

(forgot to query avail RAM on device, silly me !!)
BrunoLevy · Dec 1, 2024 · d82a5b2 · d82a5b2
1 parent 4d7591e
commit d82a5b2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/lib/geogram/NL/nl_cuda.c b/src/lib/geogram/NL/nl_cuda.c
@@ -1701,6 +1701,7 @@ static int nlCUDAFindDeviceForMatrix(NLCRSMatrix* M) {
 	    continue;
 	}
 	nlCUDACheck(CUDA()->cudaSetDevice(dev_id));
+	nlCUDACheck(CUDA()->cudaMemGetInfo(&free_RAM, &total_RAM));
 	if(free_RAM >= required_RAM) {
 	    nlCUDACheck(CUDA()->cudaSetDevice(CUDA()->main_device->devID));
 	    return dev_id;
@@ -1709,6 +1710,7 @@ static int nlCUDAFindDeviceForMatrix(NLCRSMatrix* M) {
 
     /** Oohh nooo, our matrix does not fit anywhere ! */
     nlCUDACheck(CUDA()->cudaSetDevice(CUDA()->main_device->devID));
+    nl_printf("Did not find a device with enough space for matrix");
     return -1;
 }