Add missing KOKKOS_INLINE_FUNCTION (ae765f9d) · Commits · 郑智淋 / lammps

src/KOKKOS/fix_freeze_kokkos.cpp

+1 −0

Original line number	Diff line number	Diff line
		@@ -102,6 +102,7 @@ double FixFreezeKokkos<DeviceType>::compute_vector(int n)
		/* ---------------------------------------------------------------------- */

		template<class DeviceType>
		KOKKOS_INLINE_FUNCTION
		void FixFreezeKokkos<DeviceType>::operator()(const int i, OriginalForce &original) const {
		if (mask[i] & groupbit) {
		original.values[0] += f(i,0);

src/KOKKOS/fix_rx_kokkos.cpp

+7 −0

Original line number	Diff line number	Diff line
		@@ -220,6 +220,7 @@ void FixRxKokkos<DeviceType>::rk4(const double t_stop, double y, double rwork,

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		void FixRxKokkos<DeviceType>::k_rk4(const double t_stop, VectorType& y, VectorType& rwork, UserDataType& userData) const
		{
		VectorType k1( rwork );
		@@ -279,6 +280,7 @@ void FixRxKokkos<DeviceType>::k_rk4(const double t_stop, VectorType& y, VectorTy

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		void FixRxKokkos<DeviceType>::k_rkf45_step (const int neq, const double h, VectorType& y, VectorType& y_out, VectorType& rwk, UserDataType& userData) const
		{
		const double c21=0.25;
		@@ -384,6 +386,7 @@ void FixRxKokkos<DeviceType>::k_rkf45_step (const int neq, const double h, Vecto

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		int FixRxKokkos<DeviceType>::k_rkf45_h0
		(const int neq, const double t, const double t_stop,
		const double hmin, const double hmax,
		@@ -479,6 +482,7 @@ int FixRxKokkos<DeviceType>::k_rkf45_h0

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		void FixRxKokkos<DeviceType>::k_rkf45(const int neq, const double t_stop, VectorType& y, VectorType& rwork, UserDataType& userData, CounterType& counter) const
		{
		// Rounding coefficient.
		@@ -1049,6 +1053,7 @@ int FixRxKokkos<DeviceType>::rhs_sparse(double t, const double y, double dydt,

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		int FixRxKokkos<DeviceType>::k_rhs(double t, const VectorType& y, VectorType& dydt, UserDataType& userData) const
		{
		// Use the sparse format instead.
		@@ -1062,6 +1067,7 @@ int FixRxKokkos<DeviceType>::k_rhs(double t, const VectorType& y, VectorType& dy

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		int FixRxKokkos<DeviceType>::k_rhs_dense(double t, const VectorType& y, VectorType& dydt, UserDataType& userData) const
		{
		#define rxnRateLaw (userData.rxnRateLaw)
		@@ -1103,6 +1109,7 @@ int FixRxKokkos<DeviceType>::k_rhs_dense(double t, const VectorType& y, VectorTy

		template <typename DeviceType>
		template <typename VectorType, typename UserDataType>
		KOKKOS_INLINE_FUNCTION
		int FixRxKokkos<DeviceType>::k_rhs_sparse(double t, const VectorType& y, VectorType& dydt, UserDataType& userData) const
		{
		#define kFor (userData.kFor)

src/KOKKOS/npair_ssa_kokkos.cpp

+2 −0

Original line number	Diff line number	Diff line
		@@ -525,6 +525,7 @@ fprintf(stdout, "Fina%03d %6d inum %6d gnum, total used %6d, allocated %6d\n"


		template<class DeviceType>
		KOKKOS_INLINE_FUNCTION
		void NPairSSAKokkosExecute<DeviceType>::build_locals_onePhase(const bool firstTry, int me, int workPhase) const
		{
		const typename ArrayTypes<DeviceType>::t_int_1d_const_um stencil = d_stencil;
		@@ -661,6 +662,7 @@ fprintf(stdout, "Phas%03d phase %3d used %6d inums, workItems = %3d, skipped = %


		template<class DeviceType>
		KOKKOS_INLINE_FUNCTION
		void NPairSSAKokkosExecute<DeviceType>::build_ghosts_onePhase(int workPhase) const
		{
		const typename ArrayTypes<DeviceType>::t_int_1d_const_um stencil = d_stencil;

Admin message