pair_yukawa_gpu.h
9bdacec22fa6lammm-devel

pair_yukawa_gpu.h
View Options

			/* ----------------------------------------------------------------------
			LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
			http://lammps.sandia.gov, Sandia National Laboratories
			Steve Plimpton, sjplimp@sandia.gov

			Copyright (2003) Sandia Corporation. Under the terms of Contract
			DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
			certain rights in this software. This software is distributed under
			the GNU General Public License.

			See the README file in the top-level LAMMPS directory.
			------------------------------------------------------------------------- */

			#ifdef PAIR_CLASS

			PairStyle(yukawa/gpu,PairYukawaGPU)

			#else

			#ifndef LMP_PAIR_YUKAWA_GPU_H
			#define LMP_PAIR_YUKAWA_GPU_H

			#include "pair_yukawa.h"

			namespace LAMMPS_NS {

			class PairYukawaGPU : public PairYukawa {
			public:
			PairYukawaGPU(LAMMPS *lmp);
			~PairYukawaGPU();
			void cpu_compute(int, int, int, int, int , int , int **);
			void compute(int, int);
			void init_style();
			double memory_usage();

			enum { GPU_FORCE, GPU_NEIGH, GPU_HYB_NEIGH };

			private:
			int gpu_mode;
			double cpu_time;
			int *gpulist;
			};

			}
			#endif
			#endif

			/* ERROR/WARNING messages:

			E: Out of memory on GPGPU

			GPU memory is limited. Reduce the size of the problem or increase the
			number of GPUs.

			E: Cannot use newton pair with yukawa/gpu pair style

			Self-explanatory.

			*/