@@ -2445,14 +2445,18 @@ qmckl_exit_code qmckl_compute_ee_distance_rescaled_hpc (
2445
2445
{
2446
2446
#endif
2447
2447
qmckl_exit_code rc = QMCKL_SUCCESS;
2448
+ #ifdef HAVE_OPENMP
2448
2449
#pragma omp for
2450
+ #endif
2449
2451
for (int64_t k=0 ; k<walk_num ; ++k)
2450
2452
{
2451
2453
rc |= qmckl_distance_rescaled(context, 'T', 'T', elec_num, elec_num,
2452
2454
&(coord[k*elec_num]), sze, &(coord[k*elec_num]), sze,
2453
2455
&(ee_distance_rescaled[k*elec_num2]), elec_num, rescale_factor_ee);
2454
2456
}
2457
+ #ifdef HAVE_OPENMP
2455
2458
#pragma omp critical
2459
+ #endif
2456
2460
result |= rc;
2457
2461
#ifdef HAVE_OPENMP
2458
2462
}
@@ -2802,10 +2806,14 @@ qmckl_exit_code qmckl_compute_ee_distance_rescaled_gl_hpc (
2802
2806
const int64_t elec_num2= elec_num*elec_num*4;
2803
2807
2804
2808
qmckl_exit_code result = QMCKL_SUCCESS;
2809
+ #ifdef HAVE_OPENMP
2805
2810
#pragma omp parallel
2811
+ #endif
2806
2812
{
2807
2813
qmckl_exit_code rc = QMCKL_SUCCESS;
2814
+ #ifdef HAVE_OPENMP
2808
2815
#pragma omp for
2816
+ #endif
2809
2817
for (int64_t k=0 ; k<walk_num ; ++k)
2810
2818
{
2811
2819
rc |= qmckl_distance_rescaled_gl(context, 'T', 'T', elec_num, elec_num,
@@ -2814,7 +2822,9 @@ qmckl_exit_code qmckl_compute_ee_distance_rescaled_gl_hpc (
2814
2822
&(ee_distance_rescaled_gl[k*elec_num2]), elec_num,
2815
2823
rescale_factor_ee);
2816
2824
}
2825
+ #ifdef HAVE_OPENMP
2817
2826
#pragma omp critical
2827
+ #endif
2818
2828
result |= rc;
2819
2829
}
2820
2830
return result;
@@ -4824,10 +4834,14 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_hpc (
4824
4834
const int64_t sze = elec_num*walk_num;
4825
4835
4826
4836
qmckl_exit_code result = QMCKL_SUCCESS;
4837
+ #ifdef HAVE_OPENMP
4827
4838
#pragma omp parallel
4839
+ #endif
4828
4840
{
4829
4841
qmckl_exit_code rc = QMCKL_SUCCESS;
4842
+ #ifdef HAVE_OPENMP
4830
4843
#pragma omp for
4844
+ #endif
4831
4845
for (int64_t k=0 ; k<walk_num ; ++k)
4832
4846
{
4833
4847
for (int64_t a=0 ; a<nucl_num ; ++a) {
@@ -4839,7 +4853,9 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_hpc (
4839
4853
rescale_factor_en[type_nucl_vector[a]]);
4840
4854
}
4841
4855
}
4856
+ #ifdef HAVE_OPENMP
4842
4857
#pragma omp critical
4858
+ #endif
4843
4859
result |= rc;
4844
4860
}
4845
4861
return result;
@@ -5269,10 +5285,14 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_gl_hpc (
5269
5285
const int64_t sze = elec_num*walk_num;
5270
5286
5271
5287
qmckl_exit_code result = QMCKL_SUCCESS;
5288
+ #ifdef HAVE_OPENMP
5272
5289
#pragma omp parallel
5290
+ #endif
5273
5291
{
5274
5292
qmckl_exit_code rc = QMCKL_SUCCESS;
5293
+ #ifdef HAVE_OPENMP
5275
5294
#pragma omp for
5295
+ #endif
5276
5296
for (int64_t k=0 ; k<walk_num ; ++k)
5277
5297
{
5278
5298
for (int64_t a=0 ; a<nucl_num ; ++a) {
@@ -5284,7 +5304,9 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_gl_hpc (
5284
5304
rescale_factor_en[type_nucl_vector[a]]);
5285
5305
}
5286
5306
}
5307
+ #ifdef HAVE_OPENMP
5287
5308
#pragma omp critical
5309
+ #endif
5288
5310
result |= rc;
5289
5311
}
5290
5312
return result;
@@ -7588,7 +7610,9 @@ qmckl_exit_code qmckl_compute_jastrow_champ_factor_een_rescaled_e_gl_hpc (
7588
7610
assert (elec_dist_gl2 != NULL);
7589
7611
assert (elec_dist_gl3 != NULL);
7590
7612
7613
+ #ifdef HAVE_OPENMP
7591
7614
#pragma omp parallel for
7615
+ #endif
7592
7616
for (int64_t nw = 0; nw < walk_num; ++nw) {
7593
7617
double rij_inv[elec_num];
7594
7618
for (int64_t j = 0; j < elec_num; ++j) {
@@ -7873,8 +7897,6 @@ print(" een_rescaled_e_gl[2, 1, 6, 2] = ",een_rescaled_e_gl[1, 0, 5, 2])
7873
7897
&(temp_coord[0][0][0]),
7874
7898
walk_num*3*elec_num);
7875
7899
assert(rc == QMCKL_SUCCESS);
7876
- rc = qmckl_context_touch(context);
7877
- assert(rc == QMCKL_SUCCESS);
7878
7900
7879
7901
// Call the provided function
7880
7902
rc = qmckl_get_jastrow_champ_een_distance_rescaled_e(context,
@@ -7895,7 +7917,7 @@ print(" een_rescaled_e_gl[2, 1, 6, 2] = ",een_rescaled_e_gl[1, 0, 5, 2])
7895
7917
}
7896
7918
for (int64_t nw=0 ; nw<walk_num ; nw++) {
7897
7919
temp_coord[nw][i][k] = elec_coord[nw][i][k];
7898
- }
7920
+ }
7899
7921
}
7900
7922
}
7901
7923
0 commit comments