add check for gpu device and fail fast if not available

cvxgrp · Oct 27, 2021 · a0c283b · a0c283b
1 parent 5be0e16
commit a0c283b
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 11 deletions.
diff --git a/linsys/gpu/indirect/private.c b/linsys/gpu/indirect/private.c
@@ -205,9 +205,20 @@ ScsLinSysWork *SCS(init_lin_sys_work)(const ScsMatrix *A, const ScsMatrix *P,
   cudaError_t err;
   scs_int i;
   csc *P_full;
-  ScsLinSysWork *p = (ScsLinSysWork *)scs_calloc(1, sizeof(ScsLinSysWork));
-  ScsGpuMatrix *Ag = (ScsGpuMatrix *)scs_calloc(1, sizeof(ScsGpuMatrix));
+  ScsLinSysWork *p = SCS_NULL;
+  ScsGpuMatrix *Ag = SCS_NULL;
   ScsGpuMatrix *Pg = SCS_NULL;
+  int device_count;
+
+  err = cudaGetDeviceCount(&device_count);
+  if (err > 0) {
+    scs_printf("cudaError: %i (100 indicates no device)\n", (int)err);
+    return SCS_NULL;
+  }
+
+  p = (ScsLinSysWork *)scs_calloc(1, sizeof(ScsLinSysWork));
+  Ag = (ScsGpuMatrix *)scs_calloc(1, sizeof(ScsGpuMatrix));
+
 
   p->A = A;
   p->P = P;

diff --git a/src/scs.c b/src/scs.c
@@ -841,15 +841,6 @@ static ScsWork *init_work(const ScsData *d, const ScsCone *k,
     w->cone_boundaries = SCS_NULL;
     w->scal = SCS_NULL;
   }
-  if (!(w->cone_work = SCS(init_cone)(k, w->scal, w->m))) {
-    scs_printf("ERROR: init_cone failure\n");
-    return SCS_NULL;
-  }
-  if (!(w->p =
-            SCS(init_lin_sys_work)(w->A, w->P, w->rho_y_vec, w->stgs->rho_x))) {
-    scs_printf("ERROR: init_lin_sys_work failure\n");
-    return SCS_NULL;
-  }
   /* Acceleration */
   w->rejected_accel_steps = 0;
   w->accepted_accel_steps = 0;
@@ -869,6 +860,17 @@ static ScsWork *init_work(const ScsData *d, const ScsCone *k,
   } else {
     w->accel = SCS_NULL;
   }
+  if (!(w->cone_work = SCS(init_cone)(k, w->scal, w->m))) {
+    scs_printf("ERROR: init_cone failure\n");
+    SCS(finish)(w);
+    return SCS_NULL;
+  }
+  if (!(w->p =
+            SCS(init_lin_sys_work)(w->A, w->P, w->rho_y_vec, w->stgs->rho_x))) {
+    scs_printf("ERROR: init_lin_sys_work failure\n");
+    SCS(finish)(w);
+    return SCS_NULL;
+  }
   return w;
 }