Removing unnecessary calls to sync/modified in AtomVecSphereKokkos. (0d1e55c9) · Commits · 郑智淋 / lammps

src/KOKKOS/atom_vec_sphere_kokkos.cpp

+5 −10

Original line number	Diff line number	Diff line
		@@ -549,7 +549,6 @@ void AtomVecSphereKokkos::unpack_comm_kokkos(
		return;
		}
		if(commKK->forward_comm_on_host) {
		sync(Host,X_MASK\|RADIUS_MASK\|RMASS_MASK);
		modified(Host,X_MASK\|RADIUS_MASK\|RMASS_MASK);
		struct AtomVecSphereKokkos_UnpackComm<LMPHostType> f(
		atomKK->k_x,
		@@ -557,7 +556,6 @@ void AtomVecSphereKokkos::unpack_comm_kokkos(
		buf,first);
		Kokkos::parallel_for(n,f);
		} else {
		sync(Device,X_MASK\|RADIUS_MASK\|RMASS_MASK);
		modified(Device,X_MASK\|RADIUS_MASK\|RMASS_MASK);
		struct AtomVecSphereKokkos_UnpackComm<LMPDeviceType> f(
		atomKK->k_x,
		@@ -926,7 +924,6 @@ int AtomVecSphereKokkos::pack_reverse_hybrid(int n, int first, double *buf)
		void AtomVecSphereKokkos::unpack_reverse(int n, int list, double buf)
		{
		if(n > 0) {
		sync(Host,F_MASK\|TORQUE_MASK);
		modified(Host,F_MASK\|TORQUE_MASK);
		}

		@@ -947,7 +944,6 @@ void AtomVecSphereKokkos::unpack_reverse(int n, int list, double buf)
		int AtomVecSphereKokkos::unpack_reverse_hybrid(int n, int list, double buf)
		{
		if(n > 0) {
		sync(Host,TORQUE_MASK);
		modified(Host,TORQUE_MASK);
		}

		@@ -1276,11 +1272,7 @@ struct AtomVecSphereKokkos_UnpackBorder {

		void AtomVecSphereKokkos::unpack_border_kokkos(const int &n, const int &first,
		const DAT::tdual_xfloat_2d &buf,ExecutionSpace space) {
		modified(space,X_MASK\|TAG_MASK\|TYPE_MASK\|MASK_MASK\|
		RADIUS_MASK\|RMASS_MASK);
		while (first+n >= nmax) grow(0);
		modified(space,X_MASK\|TAG_MASK\|TYPE_MASK\|MASK_MASK\|
		RADIUS_MASK\|RMASS_MASK);
		if(space==Host) {
		struct AtomVecSphereKokkos_UnpackBorder<LMPHostType> f(buf.view<LMPHostType>(),
		h_x,h_tag,h_type,h_mask,
		@@ -1294,6 +1286,9 @@ void AtomVecSphereKokkos::unpack_border_kokkos(const int &n, const int &first,
		first);
		Kokkos::parallel_for(n,f);
		}

		modified(space,X_MASK\|TAG_MASK\|TYPE_MASK\|MASK_MASK\|
		RADIUS_MASK\|RMASS_MASK);
		}

		/* ---------------------------------------------------------------------- */
		@@ -1771,10 +1766,8 @@ void AtomVecSphereKokkos::create_atom(int itype, double *coord)
		{
		int nlocal = atom->nlocal;
		if (nlocal == nmax) {
		atomKK->modified(Host,ALL_MASK);
		grow(0);
		}
		atomKK->modified(Host,ALL_MASK);

		h_tag[nlocal] = 0;
		h_type[nlocal] = itype;
		@@ -1794,6 +1787,8 @@ void AtomVecSphereKokkos::create_atom(int itype, double *coord)
		h_omega(nlocal,1) = 0.0;
		h_omega(nlocal,2) = 0.0;

		atomKK->modified(Host,ALL_MASK);

		atom->nlocal++;
		}

Admin message