Skip to content
Snippets Groups Projects
Commit ac418ec4 authored by Giovanni Lacopo's avatar Giovanni Lacopo
Browse files

Tests on GPU stacking

parent 3835bd76
Branches
Tags
No related merge requests found
......@@ -72,7 +72,8 @@ typedef struct {
double compose; //
double phase; //
double write; //
double total; } timing_t;
double total;
double offload;} timing_t;
extern timing_t timing_wt; // wall-clock process timing, at Task 0
......
......@@ -420,8 +420,9 @@ void wstack(
}
}
#ifdef ACCOMP
#pragma omp target exit data map(delete:uu[0:num_points], vv[0:num_points], ww[0:num_points], vis_real[0:Nvis], vis_img[0:Nvis], weight[0:Nvis/freq_per_chan], grid[0:2*num_w_planes*grid_size_x*grid_size_y])
#endif
// End switch between CUDA and CPU versions
#endif
//for (int i=0; i<100000; i++)printf("%f\n",grid[i]);
......
......@@ -82,6 +82,9 @@ void phase_correction(
int);
#ifdef ACCOMP
#pragma omp declare target (gauss_kernel_norm)
#endif
#ifdef __CUDACC__
extern "C"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment