Skip to content

Commit 85e1261

Browse files
committed
pragmas
1 parent 71de52c commit 85e1261

File tree

1 file changed

+25
-3
lines changed

1 file changed

+25
-3
lines changed

org/qmckl_jastrow_champ.org

Lines changed: 25 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2445,14 +2445,18 @@ qmckl_exit_code qmckl_compute_ee_distance_rescaled_hpc (
24452445
{
24462446
#endif
24472447
qmckl_exit_code rc = QMCKL_SUCCESS;
2448+
#ifdef HAVE_OPENMP
24482449
#pragma omp for
2450+
#endif
24492451
for (int64_t k=0 ; k<walk_num ; ++k)
24502452
{
24512453
rc |= qmckl_distance_rescaled(context, 'T', 'T', elec_num, elec_num,
24522454
&(coord[k*elec_num]), sze, &(coord[k*elec_num]), sze,
24532455
&(ee_distance_rescaled[k*elec_num2]), elec_num, rescale_factor_ee);
24542456
}
2457+
#ifdef HAVE_OPENMP
24552458
#pragma omp critical
2459+
#endif
24562460
result |= rc;
24572461
#ifdef HAVE_OPENMP
24582462
}
@@ -2802,10 +2806,14 @@ qmckl_exit_code qmckl_compute_ee_distance_rescaled_gl_hpc (
28022806
const int64_t elec_num2= elec_num*elec_num*4;
28032807

28042808
qmckl_exit_code result = QMCKL_SUCCESS;
2809+
#ifdef HAVE_OPENMP
28052810
#pragma omp parallel
2811+
#endif
28062812
{
28072813
qmckl_exit_code rc = QMCKL_SUCCESS;
2814+
#ifdef HAVE_OPENMP
28082815
#pragma omp for
2816+
#endif
28092817
for (int64_t k=0 ; k<walk_num ; ++k)
28102818
{
28112819
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 (
28142822
&(ee_distance_rescaled_gl[k*elec_num2]), elec_num,
28152823
rescale_factor_ee);
28162824
}
2825+
#ifdef HAVE_OPENMP
28172826
#pragma omp critical
2827+
#endif
28182828
result |= rc;
28192829
}
28202830
return result;
@@ -4824,10 +4834,14 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_hpc (
48244834
const int64_t sze = elec_num*walk_num;
48254835

48264836
qmckl_exit_code result = QMCKL_SUCCESS;
4837+
#ifdef HAVE_OPENMP
48274838
#pragma omp parallel
4839+
#endif
48284840
{
48294841
qmckl_exit_code rc = QMCKL_SUCCESS;
4842+
#ifdef HAVE_OPENMP
48304843
#pragma omp for
4844+
#endif
48314845
for (int64_t k=0 ; k<walk_num ; ++k)
48324846
{
48334847
for (int64_t a=0 ; a<nucl_num ; ++a) {
@@ -4839,7 +4853,9 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_hpc (
48394853
rescale_factor_en[type_nucl_vector[a]]);
48404854
}
48414855
}
4856+
#ifdef HAVE_OPENMP
48424857
#pragma omp critical
4858+
#endif
48434859
result |= rc;
48444860
}
48454861
return result;
@@ -5269,10 +5285,14 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_gl_hpc (
52695285
const int64_t sze = elec_num*walk_num;
52705286

52715287
qmckl_exit_code result = QMCKL_SUCCESS;
5288+
#ifdef HAVE_OPENMP
52725289
#pragma omp parallel
5290+
#endif
52735291
{
52745292
qmckl_exit_code rc = QMCKL_SUCCESS;
5293+
#ifdef HAVE_OPENMP
52755294
#pragma omp for
5295+
#endif
52765296
for (int64_t k=0 ; k<walk_num ; ++k)
52775297
{
52785298
for (int64_t a=0 ; a<nucl_num ; ++a) {
@@ -5284,7 +5304,9 @@ qmckl_exit_code qmckl_compute_en_distance_rescaled_gl_hpc (
52845304
rescale_factor_en[type_nucl_vector[a]]);
52855305
}
52865306
}
5307+
#ifdef HAVE_OPENMP
52875308
#pragma omp critical
5309+
#endif
52885310
result |= rc;
52895311
}
52905312
return result;
@@ -7588,7 +7610,9 @@ qmckl_exit_code qmckl_compute_jastrow_champ_factor_een_rescaled_e_gl_hpc (
75887610
assert (elec_dist_gl2 != NULL);
75897611
assert (elec_dist_gl3 != NULL);
75907612

7613+
#ifdef HAVE_OPENMP
75917614
#pragma omp parallel for
7615+
#endif
75927616
for (int64_t nw = 0; nw < walk_num; ++nw) {
75937617
double rij_inv[elec_num];
75947618
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])
78737897
&(temp_coord[0][0][0]),
78747898
walk_num*3*elec_num);
78757899
assert(rc == QMCKL_SUCCESS);
7876-
rc = qmckl_context_touch(context);
7877-
assert(rc == QMCKL_SUCCESS);
78787900

78797901
// Call the provided function
78807902
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])
78957917
}
78967918
for (int64_t nw=0 ; nw<walk_num ; nw++) {
78977919
temp_coord[nw][i][k] = elec_coord[nw][i][k];
7898-
}
7920+
}
78997921
}
79007922
}
79017923

0 commit comments

Comments
 (0)