|
@@ -25,7 +25,8 @@
|
|
|
* code to bootstrap the factorization
|
|
|
* and construct the DAG
|
|
|
*/
|
|
|
-static void dw_cholesky(unsigned nblocks, unsigned size, unsigned ld, float *matA[nblocks][nblocks])
|
|
|
+static void dw_cholesky(unsigned nblocks, unsigned size, unsigned ld,
|
|
|
+ float matA[nblocks][nblocks][size/nblocks * size/nblocks])
|
|
|
{
|
|
|
struct timeval start;
|
|
|
struct timeval end;
|
|
@@ -36,7 +37,7 @@ static void dw_cholesky(unsigned nblocks, unsigned size, unsigned ld, float *mat
|
|
|
|
|
|
for(x = 0; x < nblocks ; x++) {
|
|
|
for (y = 0; y < nblocks; y++) {
|
|
|
-#pragma starpu register matA[x][y] size/nblocks*size/nblocks
|
|
|
+#pragma starpu register matA[x][y]
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -112,14 +113,13 @@ int main(int argc, char **argv)
|
|
|
|
|
|
starpu_helper_cublas_init();
|
|
|
|
|
|
- float *bmat[nblocks][nblocks] __heap;
|
|
|
+ float bmat[nblocks][nblocks][BLOCKSIZE * BLOCKSIZE] __heap;
|
|
|
|
|
|
unsigned i,j,x,y;
|
|
|
for(x=0 ; x<nblocks ; x++)
|
|
|
{
|
|
|
for(y=0 ; y<nblocks ; y++)
|
|
|
{
|
|
|
- starpu_malloc((void **)&bmat[x][y], BLOCKSIZE*BLOCKSIZE*sizeof(float));
|
|
|
for (i = 0; i < BLOCKSIZE; i++)
|
|
|
{
|
|
|
for (j = 0; j < BLOCKSIZE; j++)
|
|
@@ -247,14 +247,6 @@ int main(int argc, char **argv)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- for(x=0 ; x<nblocks ; x++)
|
|
|
- {
|
|
|
- for(y=0 ; y<nblocks ; y++)
|
|
|
- {
|
|
|
- starpu_free((void *)bmat[x][y]);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
starpu_helper_cublas_shutdown();
|
|
|
#pragma starpu shutdown
|
|
|
|