p61743098
/
sagecal
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bug fix

This commit is contained in:
Sarod Yatawatta 2017-11-27 20:30:39 +01:00
parent 15ec822ec4
commit 545588bcc0
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/lib/predict_withbeam_gpu.c
View File

@ -173,6 +173,8 @@ precalcoh_threadfn(void *data) {
err=cudaSetDevice(card);
checkCudaError(err,__FILE__,__LINE__);
err=cudaSetDeviceFlags(cudaDeviceLmemResizeToMax);
checkCudaError(err,__FILE__,__LINE__);
/* make sure enough heap memory is available for shapelet computations */
size_t plim;
@ -592,6 +594,7 @@ precalcoh_threadfn(void *data) {
/* reset error state */
err=cudaGetLastError();
return NULL;
}
@ -765,6 +768,7 @@ precalculate_coherencies_withbeam_gpu(double *u, double *v, double *w, complex d
free(threaddata1);
pthread_attr_destroy(&attr);
free(th_array);
return 0;
}
@ -788,6 +792,8 @@ predictvis_threadfn(void *data) {
err=cudaSetDevice(card);
checkCudaError(err,__FILE__,__LINE__);
err=cudaSetDeviceFlags(cudaDeviceLmemResizeToMax);
checkCudaError(err,__FILE__,__LINE__);
/* make sure enough heap memory is available for shapelet computations */
size_t plim;
@ -1209,6 +1215,7 @@ predictvis_threadfn(void *data) {
}
/* reset error state */
err=cudaGetLastError();
return NULL;
@ -1358,6 +1365,8 @@ residual_threadfn(void *data) {
err=cudaSetDevice(card);
checkCudaError(err,__FILE__,__LINE__);
err=cudaSetDeviceFlags(cudaDeviceLmemResizeToMax);
checkCudaError(err,__FILE__,__LINE__);
/* make sure enough heap memory is available for shapelet computations */
size_t plim;
@ -1792,6 +1801,7 @@ residual_threadfn(void *data) {
}
/* reset error state */
err=cudaGetLastError();
return NULL;
@ -1815,6 +1825,8 @@ correct_threadfn(void *data) {
err=cudaSetDevice(card);
checkCudaError(err,__FILE__,__LINE__);
err=cudaSetDeviceFlags(cudaDeviceLmemResizeToMax);
checkCudaError(err,__FILE__,__LINE__);
double *xd;
baseline_t *barrd;
@ -1831,7 +1843,7 @@ correct_threadfn(void *data) {
/* copy with right offset */
err=cudaMemcpy(barrd, &(t->barr[t->boff]), t->Nb*sizeof(baseline_t), cudaMemcpyHostToDevice);
checkCudaError(err,__FILE__,__LINE__);
err=cudaMemcpy(pd, t->pinv, t->N*8*t->Nf*sizeof(double), cudaMemcpyHostToDevice);
err=cudaMemcpy(pd, t->pinv, t->N*8*t->nchunk*sizeof(double), cudaMemcpyHostToDevice);
checkCudaError(err,__FILE__,__LINE__);
for (cf=0; cf<t->Nf; cf++) {
@ -1855,6 +1867,7 @@ correct_threadfn(void *data) {
err=cudaFree(pd);
checkCudaError(err,__FILE__,__LINE__);
/* reset error state */
err=cudaGetLastError();
return NULL;
@ -2071,5 +2084,6 @@ double ph_ra0, double ph_dec0, double ph_freq0, double *longitude, double *latit
pthread_attr_destroy(&attr);
destroy_task_hist(&thst);
return 0;
}