Skip to content

Commit 67f916f

Browse files
committed
Fix undefined on device.
1 parent 330aabc commit 67f916f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/mam4xx/ndrop.hpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1212,7 +1212,7 @@ void update_from_explmix(
12121212
Kokkos::parallel_reduce(
12131213
Kokkos::TeamVectorRange(team, top_lev, pver_loc),
12141214
[&](int k, Real &min_val) {
1215-
const int kp1 = haero::min(k + 1, pver - 1);
1215+
const int kp1 = haero::min(k + 1, pver_loc - 1);
12161216
const int km1 = haero::max(k - 1, top_lev);
12171217
// maximum overlap assumption
12181218
if (cldn(kp1) > overlap_cld_thresh) {
@@ -1285,11 +1285,11 @@ void update_from_explmix(
12851285
nsav = nnew;
12861286
nnew = ntemp;
12871287
}
1288-
Kokkos::parallel_for(Kokkos::TeamVectorRange(team, top_lev, pver),
1288+
Kokkos::parallel_for(Kokkos::TeamVectorRange(team, top_lev, pver_loc),
12891289
[&](int k) { qncld(k) = qcld(k); });
12901290
Kokkos::parallel_for(
1291-
Kokkos::TeamVectorRange(team, top_lev, pver), [&](int k) {
1292-
const int kp1 = haero::min(k + 1, pver - 1);
1291+
Kokkos::TeamVectorRange(team, top_lev, pver_loc), [&](int k) {
1292+
const int kp1 = haero::min(k + 1, pver_loc - 1);
12931293
const int km1 = haero::max(k - 1, top_lev);
12941294
const View1D &raercol_km1_nsav = raercol[km1][nsav];
12951295
const View1D &raercol_k_nsav = raercol[k][nsav];
@@ -1304,7 +1304,7 @@ void update_from_explmix(
13041304
// k+1
13051305
// srcn(:)=srcn(:)+nact(:,m)*(raercol(:,mm,nsav))
13061306
Real srcn = zero;
1307-
if (k < pver - 1) {
1307+
if (k < pver_loc - 1) {
13081308
for (int imode = 0; imode < ntot_amode; imode++) {
13091309
const int mm = mam_idx[imode][0] - 1;
13101310
srcn += nact(k, imode) * raercol_kp1_nsav(mm);
@@ -1337,7 +1337,7 @@ void update_from_explmix(
13371337
for (int lspec = 0; lspec < nspec_amode[imode] + 1; lspec++) {
13381338
const int mm = mam_idx[imode][lspec] - 1;
13391339
Real source = 0;
1340-
if (k < pver - 1) {
1340+
if (k < pver_loc - 1) {
13411341
const Real act = lspec ? mact(k, imode) : nact(k, imode);
13421342
source = act * raercol_kp1_nsav(mm);
13431343
} else {

0 commit comments

Comments
 (0)