Browse Source

small modif

Andra Hugo 14 years ago
parent
commit
86d9ceeff8
1 changed files with 3 additions and 2 deletions
  1. 3 2
      examples/cholesky_2ctxs/cholesky_2ctxs.c

+ 3 - 2
examples/cholesky_2ctxs/cholesky_2ctxs.c

@@ -59,6 +59,7 @@ void cholesky_vs_cholesky(params *p1, params *p2, params *p3,
 
   int ncpus1 = cpu1 + gpu + gpu1;
   int ncpus2 = cpu2 + gpu + gpu2;
+  unsigned n_all_gpus = gpu + gpu1 + gpu2;
 
   /* 2 cholesky in different ctxs */
   starpu_init(NULL);
@@ -80,7 +81,7 @@ void cholesky_vs_cholesky(params *p1, params *p2, params *p3,
       //printf("%d ", i);
     }
 
-  for(i = 3; i < 3 + cpu1; i++)
+  for(i = n_all_gpus; i < n_all_gpus + cpu1; i++)
     {
       procs[k++] = i;
       //printf("%d ", i);
@@ -106,7 +107,7 @@ void cholesky_vs_cholesky(params *p1, params *p2, params *p3,
     //    printf("%d ", i);
   }
 
-  for(i = 3  + cpu1; i < 3 + cpu1 + cpu2; i++){
+  for(i = n_all_gpus  + cpu1; i < n_all_gpus + cpu1 + cpu2; i++){
     procs2[k++] = i;
     //    printf("%d ", i);
   }