pair_lj_sdk_gpu.h
No OneTemporary
Actions

Subscribers

None

File Metadata

	/* ----------------------------------------------------------------------
	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(lj/sdk/gpu,PairLJSDKGPU)
	PairStyle(cg/cmm/gpu,PairLJSDKGPU)

	#else

	#ifndef LMP_PAIR_LJ_SDK_GPU_H
	#define LMP_PAIR_LJ_SDK_GPU_H

	#include "pair_lj_sdk.h"

	namespace LAMMPS_NS {

	class PairLJSDKGPU : public PairLJSDK {
	public:
	PairLJSDKGPU(LAMMPS *lmp);
	~PairLJSDKGPU();
	template <int, int>
	void cpu_compute(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