File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed
Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -192,21 +192,23 @@ void MetatomicSystemAdaptorKokkos<DeviceType>::setup_neighbors_kk(metatomic_torc
192192 processed_all_pairs.template modify <LMPHostType>();
193193 processed_all_pairs.template sync <LMPDeviceType>();
194194
195+ auto d_numneigh = list->d_numneigh ;
196+ auto d_ilist = list->d_ilist ;
197+ auto d_neighbors = list->d_neighbors ;
198+
195199 Kokkos::parallel_for (
196200 Kokkos::MDRangePolicy<DeviceType, Kokkos::Rank<2 >>(
197201 {0 , 0 },
198202 {list->inum + list->gnum , max_number_of_neighbors}
199203 ),
200204 KOKKOS_LAMBDA (size_t ii, size_t jj) {
201- if (jj >= list-> d_numneigh [ii]) {
205+ if (jj >= d_numneigh[ii]) {
202206 return ;
203207 }
204-
205- auto atom_i = list->d_ilist [ii];
208+ auto atom_i = d_ilist[ii];
206209 auto original_atom_i = d_original_id[atom_i];
207210 auto i_is_original = (atom_i == original_atom_i);
208-
209- auto atom_j = list->d_neighbors (ii, jj) & NEIGHMASK;
211+ auto atom_j = d_neighbors (ii, jj) & NEIGHMASK;
210212 auto original_atom_j = d_original_id[atom_j];
211213 auto j_is_original = (atom_j == original_atom_j);
212214
You can’t perform that action at this time.
0 commit comments