TREX/qmckl
1
0
Fork 0
mirror of https://github.com/TREX-CoE/qmckl.git synced 2024-08-16 18:38:28 +02:00
Code Issues Projects Releases Wiki Activity

Add openmp and cublas

Browse Source
This commit is contained in:
hoffer 2022-04-06 16:20:29 +02:00
parent 39bcc569e0
commit 3b5221531c
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
org/qmckl_jastrow.org
View File

@ -5528,11 +5528,11 @@ qmckl_exit_code qmckl_compute_tmp_c_cuBlas (
const int64_t bf = elec_num*nucl_num*(cord_num+1);
const int64_t cf = bf;
const double* tmp_c_gpu = malloc(sizeof(tmp_c));
#pragma omp target enter data map(alloc:een_rescaled_e[0:elec_num*elec_num*(cord_num+1)*walk_num],een_rescaled_n[0:M*N*K],tmp_c_gpu[0:sizeof(tmp_c_gpu)/sizeof(double)])
#pragma omp target enter data map(alloc:een_rescaled_e[0:elec_num*elec_num*(cord_num+1)*walk_num],een_rescaled_n[0:M*N*walk_num],tmp_c[0:elec_num*nucl_num*(cord_num+1)*cord_num*walk_num])
#pragma omp target data use_device_ptr(een_rescaled_e,een_rescaled_n,tmp_c)
{
for (int nw=0; nw < walk_num; ++nw) {
for (int i=0; i<cord_num; ++i){
@ -5563,7 +5563,7 @@ qmckl_exit_code qmckl_compute_tmp_c_cuBlas (
cublasDestroy(handle);
#pragma omp target exit data map(from:tmp_c_gpu[0:elec_num*nucl_num*(cord_num+1)*cord_num*walk_num])
#pragma omp target exit data map(from:tmp_c[0:elec_num*nucl_num*(cord_num+1)*cord_num*walk_num])
return info;