Commit 5d3eb583 authored by Giovanni La Mura's avatar Giovanni La Mura
Browse files

Add vkv and vec_vkzm vectors to mapping

parent cb308b85

src/trapping/cfrfme.cpp

+6 −2
Original line number Original line Diff line number Diff line
@@ -649,10 +649,12 @@ void map_data( 
  double *_yv, int nyv, double *_zv, int nzv, double *vec_vkzm, int jlmf, int jlml,

  double *_yv, int nyv, double *_zv, int nzv, double *vec_vkzm, int jlmf, int jlml,

  int nkv, int nlmmt

  int nkv, int nlmmt

) {

) {

  int nkvs = nkv * nkv;

#pragma omp target enter data map(to: vec_wsum[0:size_vec_wsum]) \

#pragma omp target enter data map(to: vec_wsum[0:size_vec_wsum]) \

  map(alloc: global_vec_w[0:size_global_vec_w]) \

  map(alloc: global_vec_w[0:size_global_vec_w]) \

  map(to: vec_tt1_wk[0:size_vec_tt1_wk]) \

  map(to: vec_tt1_wk[0:size_vec_tt1_wk]) \

  map(to: _xv[0:nxv], _yv[0:nyv], _zv[0:nzv])

  map(to: _xv[0:nxv], _yv[0:nyv], _zv[0:nzv]) \

  map(to: vkv[0:nkv], vec_vkzm[0:nkvs])

}

}





void offload_loop(

void offload_loop(
@@ -719,9 +721,11 @@ void unmap_data( 
  double *_yv, int nyv, double *_zv, int nzv, double *vec_vkzm, int jlmf, int jlml,

  double *_yv, int nyv, double *_zv, int nzv, double *vec_vkzm, int jlmf, int jlml,

  int nkv, int nlmmt

  int nkv, int nlmmt

) {

) {

  int nkvs = nkv * nkv;

#pragma omp target exit data map(from: vec_wsum[0:size_vec_wsum]) \

#pragma omp target exit data map(from: vec_wsum[0:size_vec_wsum]) \

  map(delete: global_vec_w[0:size_global_vec_w]) \

  map(delete: global_vec_w[0:size_global_vec_w]) \

  map(delete: vec_tt1_wk[0:size_vec_tt1_wk]) \

  map(delete: vec_tt1_wk[0:size_vec_tt1_wk]) \

  map(delete: _xv[0:nxv], _yv[0:nyv], _zv[0:nzv])

  map(delete: _xv[0:nxv], _yv[0:nyv], _zv[0:nzv]) \

  map(delete: vkv[0:nkv], vec_vkzm[0:nkvs])

}

}

#endif // USE TARGET_OFFLOAD
 
#endif // USE TARGET_OFFLOAD