lijiext
/
lammps
mirror of https://github.com/lammps/lammps.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity
lammps/lib/cuda/cuda_pair.cu

1021 lines
41 KiB
Plaintext
Raw Blame History

/* ----------------------------------------------------------------------
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
Original Version:
http://lammps.sandia.gov, Sandia National Laboratories
Steve Plimpton, sjplimp@sandia.gov
See the README file in the top-level LAMMPS directory.
-----------------------------------------------------------------------
USER-CUDA Package and associated modifications:
https://sourceforge.net/projects/lammpscuda/
Christian Trott, christian.trott@tu-ilmenau.de
Lars Winterfeld, lars.winterfeld@tu-ilmenau.de
Theoretical Physics II, University of Technology Ilmenau, Germany
See the README file in the USER-CUDA directory.
This software is distributed under the GNU General Public License.
------------------------------------------------------------------------- */
enum PAIR_FORCES {PAIR_NONE,PAIR_BORN,PAIR_BUCK,PAIR_CG_CMM,PAIR_LJ_CHARMM,PAIR_LJ_CLASS2,PAIR_LJ_CUT, PAIR_LJ_EXPAND, PAIR_LJ_GROMACS, PAIR_LJ_SMOOTH, PAIR_LJ96_CUT, PAIR_MORSE, PAIR_MORSE_R6};
enum COUL_FORCES {COUL_NONE,COUL_CHARMM,COUL_CHARMM_IMPLICIT,COUL_CUT,COUL_LONG, COUL_DEBYE, COUL_GROMACS,COUL_SPECIAL};
#define DATA_NONE 0
#define DATA_V 1
#define DATA_TAG 2
#define DATA_RMASS 4
#define DATA_MASS 8
#define DATA_TORQUE 16
#define DATA_OMEGA 32
#define DATA_RADIUS 64
#define DATA_DENSITY 128
#define DATA_MASK 256
#define DATA_V_RADIUS 512
#define DATA_OMEGA_RMASS 1024
#define SBBITS 30
#define NEIGHMASK 0x3FFFFFFF
#define MY_PREFIX cuda_pair
#define IncludeCommonNeigh
#include "cuda_shared.h"
#include "cuda_common.h"
#include "cuda_wrapper_cu.h"
#include "crm_cuda_utils.cu"
//constants used by multiple forces
//general
#define _cutsq MY_AP(cutsq)
#define _offset MY_AP(offset)
#define _special_lj MY_AP(special_lj)
#define _special_coul MY_AP(special_coul)
#define _cutsq_global MY_AP(cutsq_global)
#define _collect_forces_later MY_AP(collect_forces_later)
__device__ __constant__ X_FLOAT _cutsq[CUDA_MAX_TYPES2];
__device__ __constant__ ENERGY_FLOAT _offset[CUDA_MAX_TYPES2];
__device__ __constant__ F_FLOAT _special_lj[4];
__device__ __constant__ F_FLOAT _special_coul[4];
__device__ __constant__ X_FLOAT _cutsq_global;
__device__ __constant__ int _collect_forces_later;
__device__ __constant__ F_FLOAT MY_AP(coeff1)[CUDA_MAX_TYPES2]; //pair force coefficients in case ntypes < CUDA_MAX_TYPES (coeffs fit into constant space)
__device__ __constant__ F_FLOAT MY_AP(coeff2)[CUDA_MAX_TYPES2];
__device__ __constant__ F_FLOAT MY_AP(coeff3)[CUDA_MAX_TYPES2];
__device__ __constant__ F_FLOAT MY_AP(coeff4)[CUDA_MAX_TYPES2];
__device__ __constant__ F_FLOAT MY_AP(coeff5)[CUDA_MAX_TYPES2];
__device__ __constant__ F_FLOAT* MY_AP(coeff1_gm); //pair force coefficients in case ntypes > CUDA_MAX_TYPES (coeffs do not fit into constant space)
__device__ __constant__ F_FLOAT* MY_AP(coeff2_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff3_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff4_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff5_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff6_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff7_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff8_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff9_gm);
__device__ __constant__ F_FLOAT* MY_AP(coeff10_gm);
#define _coeff1_gm_tex MY_AP(coeff1_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff1_gm_tex;
#else
texture<int2,1> _coeff1_gm_tex;
#endif
#define _coeff2_gm_tex MY_AP(coeff2_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff2_gm_tex;
#else
texture<int2,1> _coeff2_gm_tex;
#endif
#define _coeff3_gm_tex MY_AP(coeff3_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff3_gm_tex;
#else
texture<int2,1> _coeff3_gm_tex;
#endif
#define _coeff4_gm_tex MY_AP(coeff4_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff4_gm_tex;
#else
texture<int2,1> _coeff4_gm_tex;
#endif
#define _coeff5_gm_tex MY_AP(coeff5_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff5_gm_tex;
#else
texture<int2,1> _coeff5_gm_tex;
#endif
#define _coeff6_gm_tex MY_AP(coeff6_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff6_gm_tex;
#else
texture<int2,1> _coeff6_gm_tex;
#endif
#define _coeff7_gm_tex MY_AP(coeff7_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff7_gm_tex;
#else
texture<int2,1> _coeff7_gm_tex;
#endif
#define _coeff8_gm_tex MY_AP(coeff8_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff8_gm_tex;
#else
texture<int2,1> _coeff8_gm_tex;
#endif
#define _coeff9_gm_tex MY_AP(coeff9_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff9_gm_tex;
#else
texture<int2,1> _coeff9_gm_tex;
#endif
#define _coeff10_gm_tex MY_AP(coeff10_gm_tex)
#if F_PRECISION == 1
texture<float> _coeff10_gm_tex;
#else
texture<int2,1> _coeff10_gm_tex;
#endif
//if more than 5 coefficients are needed for a pair potential add them here
//coulomb
#define _cut_coulsq MY_AP(cut_coulsq)
#define _cut_coulsq_global MY_AP(cut_coulsq_global)
#define _g_ewald MY_AP(g_ewald)
#define _qqrd2e MY_AP(qqrd2e)
#define _kappa MY_AP(kappa)
__device__ __constant__ X_FLOAT _cut_coulsq[CUDA_MAX_TYPES2];
__device__ __constant__ X_FLOAT _cut_coulsq_global;
__device__ __constant__ F_FLOAT _g_ewald;
__device__ __constant__ F_FLOAT _qqrd2e;
__device__ __constant__ F_FLOAT _kappa;
//inner cutoff
#define _cut_innersq MY_AP(cut_innersq)
#define _cut_innersq_global MY_AP(cut_innersq_global)
__device__ __constant__ X_FLOAT _cut_innersq[CUDA_MAX_TYPES2];
__device__ __constant__ X_FLOAT _cut_innersq_global;
template <const PAIR_FORCES pair_type,const COUL_FORCES coul_type,const unsigned int extended_data>
__global__ void Pair_Kernel_TpA(int eflag, int vflag,int eflag_atom,int vflag_atom);
template <const PAIR_FORCES pair_type,const COUL_FORCES coul_type,const unsigned int extended_data>
__global__ void Pair_Kernel_BpA(int eflag, int vflag,int eflag_atom,int vflag_atom);
template <const PAIR_FORCES pair_type,const COUL_FORCES coul_type,const unsigned int extended_data>
__global__ void Pair_Kernel_TpA_opt(int eflag, int vflag,int eflag_atom,int vflag_atom, int comm_phase);
template <const PAIR_FORCES pair_type,const COUL_FORCES coul_type,const unsigned int extended_data>
__global__ void Pair_Kernel_BpA_opt(int eflag, int vflag,int eflag_atom,int vflag_atom, int comm_phase);
#include <stdio.h>
#include "cuda_pair_cu.h"
#include "cuda_pair_virial_kernel_nc.cu"
//Functions which are shared by pair styles
//Update Buffersize
void Cuda_UpdateBuffer(cuda_shared_data* sdata,int size)
{
CUT_CHECK_ERROR("Cuda_Pair_UpdateBuffer_AllStyles: before updateBuffer failed");
if(sdata->buffersize<size)
{
MYDBG(printf("Resizing Buffer at %p with %i kB to\n",sdata->buffer,sdata->buffersize);)
CudaWrapper_FreeCudaData(sdata->buffer,sdata->buffersize);
sdata->buffer = CudaWrapper_AllocCudaData(size);
sdata->buffersize=size;
sdata->buffer_new++;
MYDBG(printf("New buffer at %p with %i kB\n",sdata->buffer,sdata->buffersize);)
}
cudaMemcpyToSymbol(MY_CONST(buffer), & sdata->buffer, sizeof(int*) );
CUT_CHECK_ERROR("Cuda_Pair_UpdateBuffer_AllStyles failed");
}
void Cuda_Pair_UpdateNeighbor_AllStyles(cuda_shared_data* sdata, cuda_shared_neighlist* sneighlist)
{
//Neighbor
cudaMemcpyToSymbol(MY_CONST(neighbor_maxlocal) , & sneighlist->firstneigh.dim[0] , sizeof(unsigned) );
cudaMemcpyToSymbol(MY_CONST(firstneigh) , & sneighlist->firstneigh.dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(ilist) , & sneighlist->ilist .dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(inum) , & sneighlist->inum , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(numneigh) , & sneighlist->numneigh .dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(neighbors) , & sneighlist->neighbors .dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(maxneighbors) , & sneighlist->maxneighbors , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(overlap_comm) , & sdata->overlap_comm, sizeof(int) );
if(sdata->overlap_comm)
{
cudaMemcpyToSymbol(MY_CONST(numneigh_border) , & sneighlist->numneigh_border .dev_data, sizeof(int*));
cudaMemcpyToSymbol(MY_CONST(numneigh_inner) , & sneighlist->numneigh_inner .dev_data, sizeof(int*));
cudaMemcpyToSymbol(MY_CONST(neighbors_border) , & sneighlist->neighbors_border.dev_data, sizeof(int*));
cudaMemcpyToSymbol(MY_CONST(neighbors_inner) , & sneighlist->neighbors_inner .dev_data, sizeof(int*));
cudaMemcpyToSymbol(MY_CONST(ilist_border) , & sneighlist->ilist_border .dev_data, sizeof(int*));
cudaMemcpyToSymbol(MY_CONST(inum_border) , & sneighlist->inum_border .dev_data, sizeof(int*) );
}
}
//Update constants after nmax change which are generally needed by all pair styles
void Cuda_Pair_UpdateNmax_AllStyles(cuda_shared_data* sdata, cuda_shared_neighlist* sneighlist)
{
CUT_CHECK_ERROR("Cuda_Pair_UpdateNmax_AllStyles: Begin");
//System
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nmax) , & sdata->atom.nmax , sizeof(int) );
//Atom
cudaMemcpyToSymbol(MY_CONST(x) , & sdata->atom.x .dev_data, sizeof(X_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(x_type) , & sdata->atom.x_type .dev_data, sizeof(X_FLOAT4*) );
cudaMemcpyToSymbol(MY_CONST(f) , & sdata->atom.f .dev_data, sizeof(F_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(type) , & sdata->atom.type .dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(q) , & sdata->atom.q .dev_data, sizeof(F_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(tag) , & sdata->atom.tag .dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(eatom) , & sdata->atom.eatom .dev_data, sizeof(ENERGY_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(vatom) , & sdata->atom.vatom .dev_data, sizeof(ENERGY_FLOAT*) );
//Other
cudaMemcpyToSymbol(MY_CONST(debugdata) , & sdata->debugdata , sizeof(int*) );
CUT_CHECK_ERROR("Cuda_Pair_UpdateNmax_AllStyles: End");
}
//Initialisation of GPU Constants which rarely change
void Cuda_Pair_Init_AllStyles(cuda_shared_data* sdata, int ncoeff, bool need_q=false, bool use_global_params=false, bool need_innercut=false, bool need_cut=true )
{
unsigned cuda_ntypes = sdata->atom.ntypes + 1;
unsigned cuda_ntypes2 = cuda_ntypes * cuda_ntypes;
unsigned n = sizeof(F_FLOAT) * cuda_ntypes2;
unsigned nx = sizeof(X_FLOAT) * cuda_ntypes2;
//check if enough constant memory is available
if((cuda_ntypes2 > CUDA_MAX_TYPES2 )&& !use_global_params)
printf("# CUDA: Cuda_Pair_Init: you need %u types. this is more than %u "
"(assumed at compile time). re-compile with -DCUDA_MAX_TYPES_PLUS_ONE=32 "
"or ajust this in cuda_common.h\n", cuda_ntypes, CUDA_MAX_TYPES_PLUS_ONE-1);
if((cuda_ntypes2 > CUDA_MAX_TYPES2 )&& !use_global_params)
exit(0);
//type conversion of cutoffs and parameters
if(need_cut)
{
X_FLOAT cutsq[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
cutsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_global * sdata->pair.cut_global);
}
}
int cutsqdiffer=0;
X_FLOAT cutsq_global;
cutsq_global = (X_FLOAT) (sdata->pair.cut_global * sdata->pair.cut_global);
if(sdata->pair.cut)
{
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
if(sdata->pair.cut[i][j]>1e-6)
cutsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut[i][j] * sdata->pair.cut[i][j]);
else
if(sdata->pair.cut[j][i]>1e-6)
cutsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut[j][i] * sdata->pair.cut[j][i]);
if(i==1&&j==1) cutsq_global = cutsq[i * cuda_ntypes + j];
if((cutsq_global - cutsq[i * cuda_ntypes + j])*(cutsq_global - cutsq[i * cuda_ntypes + j]) > 1e-6)
cutsqdiffer++;
}
}
}
if(sdata->pair.cutsq)
{
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
if(sdata->pair.cut[i][j]>1e-6)
cutsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cutsq[i][j]);
else
if(sdata->pair.cut[j][i]>1e-6)
cutsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cutsq[j][i]);
if(i==1&&j==1) cutsq_global = cutsq[i * cuda_ntypes + j];
if((cutsq_global - cutsq[i * cuda_ntypes + j])*(cutsq_global - cutsq[i * cuda_ntypes + j]) > 1e-6)
cutsqdiffer++;
}
}
}
if(cutsqdiffer)
{
cutsq_global = -1.0;
cudaMemcpyToSymbol(MY_CONST(cutsq) , cutsq , nx );
}
cudaMemcpyToSymbol(MY_CONST(cutsq_global) ,&cutsq_global , sizeof(X_FLOAT) );
}
if(need_innercut)
{
X_FLOAT cut_innersq[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
cut_innersq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_inner_global * sdata->pair.cut_inner_global);
}
}
int cutsqdiffer=0;
X_FLOAT cut_innersq_global;
cut_innersq_global = (X_FLOAT) (sdata->pair.cut_inner_global * sdata->pair.cut_inner_global);
if(sdata->pair.cut_inner)
{
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
if(sdata->pair.cut_inner[i][j]>1e-6)
cut_innersq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_inner[i][j] * sdata->pair.cut_inner[i][j]);
else
if(sdata->pair.cut_inner[j][i]>1e-6)
cut_innersq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_inner[j][i] * sdata->pair.cut_inner[j][i]);
if(i==1&&j==1) cut_innersq_global = cut_innersq[i * cuda_ntypes + j];
if((cut_innersq_global - cut_innersq[i * cuda_ntypes + j])*(cut_innersq_global - cut_innersq[i * cuda_ntypes + j]) > 1e-6)
cutsqdiffer++;
}
}
}
if(cutsqdiffer)
{
cut_innersq_global = -1.0;
cudaMemcpyToSymbol(MY_CONST(cut_innersq) , cut_innersq , nx );
}
cudaMemcpyToSymbol(MY_CONST(cut_innersq_global) ,&cut_innersq_global , sizeof(X_FLOAT) );
}
if(need_q)
{
X_FLOAT cut_coulsq[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
cut_coulsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_coul_global * sdata->pair.cut_coul_global);
}
}
int cutsqdiffer=0;
X_FLOAT cut_coulsq_global;
cut_coulsq_global = (X_FLOAT) (sdata->pair.cut_coul_global * sdata->pair.cut_coul_global);
if(sdata->pair.cut_coulsq_global> cut_coulsq_global) cut_coulsq_global = (X_FLOAT) sdata->pair.cut_coulsq_global;
if(sdata->pair.cut_coul)
{
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
if(sdata->pair.cut_coul[i][j]>1e-6)
cut_coulsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_coul[i][j] * sdata->pair.cut_coul[i][j]);
else
if(sdata->pair.cut_coul[j][i]>1e-6)
cut_coulsq[i * cuda_ntypes + j] = (X_FLOAT) (sdata->pair.cut_coul[j][i] * sdata->pair.cut_coul[j][i]);
if(i==1&&j==1) cut_coulsq_global = cut_coulsq[i * cuda_ntypes + j];
if((cut_coulsq_global - cut_coulsq[i * cuda_ntypes + j])*(cut_coulsq_global - cut_coulsq[i * cuda_ntypes + j]) > 1e-6)
cutsqdiffer++;
}
}
}
if(cutsqdiffer)
{
cut_coulsq_global = -1.0;
cudaMemcpyToSymbol(MY_CONST(cut_coulsq) , cut_coulsq , nx );
}
cudaMemcpyToSymbol(MY_CONST(cut_coulsq_global),&cut_coulsq_global , sizeof(X_FLOAT) );
}
CUT_CHECK_ERROR("Cuda_Pair: init pre Coeff failed");
if(ncoeff>0)
{
F_FLOAT coeff1[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff1[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff1[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff1_gm) , &sdata->pair.coeff1_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy((sdata->pair.coeff1_gm.dev_data),coeff1, n,cudaMemcpyHostToDevice);
_coeff1_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff1_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff1_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff1_gm_texture_ptr;
cudaGetTextureReference(&coeff1_gm_texture_ptr, MY_CONST(coeff1_gm_tex));
CUT_CHECK_ERROR("Cuda_Pair: init Coeff0 a failed");
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
CUT_CHECK_ERROR("Cuda_Pair: init Coeff0 b failed");
cudaBindTexture(0,coeff1_gm_texture_ptr, sdata->pair.coeff1_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
CUT_CHECK_ERROR("Cuda_Pair: init Coeff0 c failed");
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
CUT_CHECK_ERROR("Cuda_Pair: init Coeff0 b-d failed");
cudaBindTexture(0,coeff1_gm_texture_ptr, sdata->pair.coeff1_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
CUT_CHECK_ERROR("Cuda_Pair: init Coeff0 c-d failed");
#endif
}
else
cudaMemcpyToSymbol(MY_AP(coeff1), coeff1 , n);
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff0 failed");
if(ncoeff>1)
{
F_FLOAT coeff2[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff2[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff2[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff2_gm) , &sdata->pair.coeff2_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff2_gm.dev_data, coeff2, n,cudaMemcpyHostToDevice);
_coeff2_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff2_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff2_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff2_gm_texture_ptr;
cudaGetTextureReference(&coeff2_gm_texture_ptr, MY_CONST(coeff2_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff2_gm_texture_ptr, sdata->pair.coeff2_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff2_gm_texture_ptr, sdata->pair.coeff2_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
else
cudaMemcpyToSymbol(MY_AP(coeff2), coeff2 , n);
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff1 failed");
if(ncoeff>2)
{
F_FLOAT coeff3[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff3[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff3[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff3_gm) , &sdata->pair.coeff3_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff3_gm.dev_data, coeff3, n,cudaMemcpyHostToDevice);
_coeff3_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff3_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff3_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff3_gm_texture_ptr;
cudaGetTextureReference(&coeff3_gm_texture_ptr, MY_CONST(coeff3_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff3_gm_texture_ptr, sdata->pair.coeff3_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff3_gm_texture_ptr, sdata->pair.coeff3_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
else
cudaMemcpyToSymbol(MY_AP(coeff3), coeff3 , n);
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff3 failed");
if(ncoeff>3)
{
F_FLOAT coeff4[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff4[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff4[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff4_gm) , &sdata->pair.coeff4_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff4_gm.dev_data, coeff4, n,cudaMemcpyHostToDevice);
_coeff4_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff4_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff4_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff4_gm_texture_ptr;
cudaGetTextureReference(&coeff4_gm_texture_ptr, MY_CONST(coeff4_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff4_gm_texture_ptr, sdata->pair.coeff4_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff4_gm_texture_ptr, sdata->pair.coeff4_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
else
cudaMemcpyToSymbol(MY_AP(coeff4), coeff4 , n);
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff4 failed");
if(ncoeff>4)
{
F_FLOAT coeff5[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff5[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff5[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff5_gm) , &sdata->pair.coeff5_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff5_gm.dev_data, coeff5, n,cudaMemcpyHostToDevice);
_coeff5_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff5_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff5_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff5_gm_texture_ptr;
cudaGetTextureReference(&coeff5_gm_texture_ptr, MY_CONST(coeff5_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff5_gm_texture_ptr, sdata->pair.coeff5_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff5_gm_texture_ptr, sdata->pair.coeff5_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
else
cudaMemcpyToSymbol(MY_AP(coeff5), coeff5 , n);
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff5 failed");
if(ncoeff>5)
{
F_FLOAT coeff6[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff6[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff6[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff6_gm) , &sdata->pair.coeff6_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff6_gm.dev_data, coeff6, n,cudaMemcpyHostToDevice);
_coeff6_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff6_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff6_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff6_gm_texture_ptr;
cudaGetTextureReference(&coeff6_gm_texture_ptr, MY_CONST(coeff6_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff6_gm_texture_ptr, sdata->pair.coeff6_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff6_gm_texture_ptr, sdata->pair.coeff6_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff6 failed");
if(ncoeff>6)
{
F_FLOAT coeff7[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff7[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff7[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff7_gm) , &sdata->pair.coeff7_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff7_gm.dev_data, coeff7, n,cudaMemcpyHostToDevice);
_coeff7_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff7_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff7_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff7_gm_texture_ptr;
cudaGetTextureReference(&coeff7_gm_texture_ptr, MY_CONST(coeff7_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff7_gm_texture_ptr, sdata->pair.coeff7_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff7_gm_texture_ptr, sdata->pair.coeff7_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff7 failed");
if(ncoeff>7)
{
F_FLOAT coeff8[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff8[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff8[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff8_gm) , &sdata->pair.coeff8_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff8_gm.dev_data, coeff8, n,cudaMemcpyHostToDevice);
_coeff8_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff8_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff8_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff8_gm_texture_ptr;
cudaGetTextureReference(&coeff8_gm_texture_ptr, MY_CONST(coeff8_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff8_gm_texture_ptr, sdata->pair.coeff8_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff8_gm_texture_ptr, sdata->pair.coeff8_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff8 failed");
if(ncoeff>8)
{
F_FLOAT coeff9[cuda_ntypes2];
for(int i=1; i<=sdata->atom.ntypes; ++i)
{
for(int j=1; j<=sdata->atom.ntypes; ++j)
{
coeff9[i * cuda_ntypes + j] = (F_FLOAT) sdata->pair.coeff9[i][j];
}
}
if(use_global_params)
{
cudaMemcpyToSymbol(MY_CONST(coeff9_gm) , &sdata->pair.coeff9_gm.dev_data , sizeof(F_FLOAT*) );
cudaMemcpy(sdata->pair.coeff9_gm.dev_data, coeff9, n,cudaMemcpyHostToDevice);
_coeff9_gm_tex.normalized = false; // access with normalized texture coordinates
_coeff9_gm_tex.filterMode = cudaFilterModePoint; // Point mode, so no
_coeff9_gm_tex.addressMode[0] = cudaAddressModeWrap; // wrap texture coordinates
const textureReference* coeff9_gm_texture_ptr;
cudaGetTextureReference(&coeff9_gm_texture_ptr, MY_CONST(coeff9_gm_tex));
#if F_PRECISION == 1
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<float>();
cudaBindTexture(0,coeff9_gm_texture_ptr, sdata->pair.coeff9_gm.dev_data, &channelDescXType, sdata->atom.nmax*sizeof(F_FLOAT));
#else
cudaChannelFormatDesc channelDescXType = cudaCreateChannelDesc<int2>();
cudaBindTexture(0,coeff9_gm_texture_ptr, sdata->pair.coeff9_gm.dev_data, &channelDescXType, sdata->atom.nmax*2*sizeof(int2));
#endif
}
}
CUT_CHECK_ERROR("Cuda_Pair: init Coeff9 failed");
F_FLOAT special_lj[4];
special_lj[0]=sdata->pair.special_lj[0];
special_lj[1]=sdata->pair.special_lj[1];
special_lj[2]=sdata->pair.special_lj[2];
special_lj[3]=sdata->pair.special_lj[3];
X_FLOAT box_size[3] =
{
sdata->domain.subhi[0] - sdata->domain.sublo[0],
sdata->domain.subhi[1] - sdata->domain.sublo[1],
sdata->domain.subhi[2] - sdata->domain.sublo[2]
};
cudaMemcpyToSymbol(MY_CONST(box_size) , box_size , sizeof(X_FLOAT)*3);
cudaMemcpyToSymbol(MY_CONST(cuda_ntypes) ,&cuda_ntypes , sizeof(unsigned) );
cudaMemcpyToSymbol(MY_CONST(special_lj) , special_lj , sizeof(F_FLOAT)*4);
cudaMemcpyToSymbol(MY_CONST(virial) ,&sdata->pair.virial.dev_data , sizeof(ENERGY_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(eng_vdwl) ,&sdata->pair.eng_vdwl.dev_data , sizeof(ENERGY_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(periodicity) , sdata->domain.periodicity , sizeof(int)*3 );
cudaMemcpyToSymbol(MY_CONST(collect_forces_later), &sdata->pair.collect_forces_later , sizeof(int) );
if(need_q)
{
F_FLOAT qqrd2e_tmp=sdata->pppm.qqrd2e;
F_FLOAT special_coul[4];
special_coul[0]=sdata->pair.special_coul[0];
special_coul[1]=sdata->pair.special_coul[1];
special_coul[2]=sdata->pair.special_coul[2];
special_coul[3]=sdata->pair.special_coul[3];
cudaMemcpyToSymbol(MY_CONST(special_coul) , special_coul , sizeof(F_FLOAT)*4);
cudaMemcpyToSymbol(MY_CONST(g_ewald) ,&sdata->pair.g_ewald , sizeof(F_FLOAT) );
cudaMemcpyToSymbol(MY_CONST(qqrd2e) ,&qqrd2e_tmp , sizeof(F_FLOAT) );
cudaMemcpyToSymbol(MY_CONST(kappa) ,&sdata->pair.kappa , sizeof(F_FLOAT) );
cudaMemcpyToSymbol(MY_CONST(eng_coul) ,&sdata->pair.eng_coul.dev_data , sizeof(ENERGY_FLOAT*) );
}
CUT_CHECK_ERROR("Cuda_Pair: init failed");
}
timespec startpairtime, endpairtime;
//Function which is called prior to kernel invocation, determins grid, Binds Textures, updates constant memory if necessary
void Cuda_Pair_PreKernel_AllStyles(cuda_shared_data* sdata, cuda_shared_neighlist* sneighlist,int eflag, int vflag, dim3& grid, dim3& threads, int& sharedperproc,bool need_q=false,int maxthreads=256)
{
if(sdata->atom.update_neigh)
Cuda_Pair_UpdateNeighbor_AllStyles(sdata,sneighlist);
if(sdata->atom.update_nmax)
Cuda_Pair_UpdateNmax_AllStyles(sdata,sneighlist);
if(sdata->atom.update_nlocal)
{
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
}
BindXTypeTexture(sdata);
if(need_q) BindQTexture(sdata);
sharedperproc=0;
if(sdata->pair.use_block_per_atom) sharedperproc+=3;
if(eflag) sharedperproc+=1;
if(need_q && eflag) sharedperproc+=1;
if(vflag) sharedperproc+=6;
int threadnum = sneighlist->inum;
if (sdata->comm.comm_phase==2)threadnum=sneighlist->inum_border2;
if(sdata->pair.use_block_per_atom) {threadnum*=64; maxthreads=64;}
int3 layout=getgrid(threadnum,sharedperproc*sizeof(ENERGY_FLOAT),maxthreads,true); //need to limit to 192 threads due to register limit
threads.x = layout.z; threads.y = 1; threads.z = 1;
grid.x = layout.x; grid.y = layout.y; grid.z = 1;
int size=(unsigned)(layout.y*layout.x)*sharedperproc*sizeof(ENERGY_FLOAT);
Cuda_UpdateBuffer(sdata,size);
if(sdata->pair.use_block_per_atom)
cudaMemset(sdata->buffer, 0, size);
sdata->pair.lastgridsize=grid.x*grid.y;
sdata->pair.n_energy_virial=sharedperproc;
if(sdata->pair.use_block_per_atom) sdata->pair.n_energy_virial-=3;
clock_gettime(CLOCK_REALTIME,&startpairtime);
MYDBG( printf("# CUDA: Cuda_Pair: kernel start eflag: %i vflag: %i config: %i %i %i %i\n",eflag,vflag,grid.x,grid.y, threads.x,sharedperproc*sizeof(ENERGY_FLOAT)*threads.x); )
}
//Function which is called after the kernel invocation, collects energy and virial
void Cuda_Pair_PostKernel_AllStyles(cuda_shared_data* sdata, dim3& grid, int& sharedperproc,int eflag, int vflag)
{
if((not sdata->pair.collect_forces_later) && (eflag||vflag))//not sdata->comm.comm_phase==2))
{
cudaThreadSynchronize();
clock_gettime(CLOCK_REALTIME,&endpairtime);
sdata->cuda_timings.pair_kernel+=
endpairtime.tv_sec-startpairtime.tv_sec+1.0*(endpairtime.tv_nsec-startpairtime.tv_nsec)/1000000000;
CUT_CHECK_ERROR("Cuda_Pair: Kernel execution failed");
if(eflag||vflag)
{
int n=grid.x*grid.y;
if(sdata->pair.use_block_per_atom)
grid.x=sharedperproc-3;
else
grid.x=sharedperproc;
grid.y=1;
dim3 threads(128,1,1);
MYDBG( printf("# CUDA: Cuda_Pair: virial compute kernel start eflag: %i vflag: %i config: %i %i %i %i\n",eflag,vflag,grid.x,grid.y, threads.x,sharedperproc*sizeof(ENERGY_FLOAT)*threads.x); )
MY_AP(PairVirialCompute_reduce)<<<grid,threads,threads.x*sizeof(ENERGY_FLOAT)>>>(n);
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair: virial compute Kernel execution failed");
}
MYDBG( printf("# CUDA: Cuda_Pair: kernel done\n"); )
}
}
#include "pair_born_coul_long_cuda.cu"
#include "pair_buck_coul_cut_cuda.cu"
#include "pair_buck_coul_long_cuda.cu"
#include "pair_buck_cuda.cu"
#include "pair_cg_cmm_cuda.cu"
#include "pair_cg_cmm_coul_cut_cuda.cu"
#include "pair_cg_cmm_coul_debye_cuda.cu"
#include "pair_cg_cmm_coul_long_cuda.cu"
#include "pair_gran_hooke_cuda.cu"
#include "pair_lj_charmm_coul_charmm_implicit_cuda.cu"
#include "pair_lj_charmm_coul_charmm_cuda.cu"
#include "pair_lj_charmm_coul_long_cuda.cu"
#include "pair_lj_class2_coul_cut_cuda.cu"
#include "pair_lj_class2_coul_long_cuda.cu"
#include "pair_lj_class2_cuda.cu"
#include "pair_lj_cut_coul_cut_cuda.cu"
#include "pair_lj_cut_coul_debye_cuda.cu"
#include "pair_lj_cut_coul_long_cuda.cu"
#include "pair_lj_cut_cuda.cu"
#include "pair_lj_cut_experimental_cuda.cu"
#include "pair_lj_expand_cuda.cu"
#include "pair_lj_gromacs_cuda.cu"
#include "pair_lj_gromacs_coul_gromacs_cuda.cu"
#include "pair_lj_smooth_cuda.cu"
#include "pair_lj96_cut_cuda.cu"
#include "pair_morse_coul_long_cuda.cu"
#include "pair_morse_cuda.cu"
#include "pair_eam_cuda.cu"
#include "cuda_pair_kernel.cu"
#include "pair_manybody_const.h"
#include "pair_tersoff_cuda.cu"
#include "pair_sw_cuda.cu"
void Cuda_Pair_UpdateNmax(cuda_shared_data* sdata)
{
CUT_CHECK_ERROR("Cuda_Pair: before updateNmax failed");
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nmax) , & sdata->atom.nmax , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(type) , & sdata->atom.type .dev_data, sizeof(int*) );
cudaMemcpyToSymbol(MY_CONST(x) , & sdata->atom.x .dev_data, sizeof(X_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(x_type) , & sdata->atom.x_type .dev_data, sizeof(X_FLOAT4*) );
cudaMemcpyToSymbol(MY_CONST(xhold) , & sdata->atom.xhold .dev_data, sizeof(X_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(v) , & sdata->atom.v .dev_data, sizeof(V_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(radius) , & sdata->atom.radius .dev_data, sizeof(X_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(v_radius) , & sdata->atom.v_radius .dev_data, sizeof(V_FLOAT4*) );
cudaMemcpyToSymbol(MY_CONST(omega) , & sdata->atom.omega .dev_data, sizeof(V_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(rmass) , & sdata->atom.rmass .dev_data, sizeof(V_FLOAT*) );
cudaMemcpyToSymbol(MY_CONST(omega_rmass),& sdata->atom.omega_rmass.dev_data, sizeof(V_FLOAT4*) );
CUT_CHECK_ERROR("Cuda_Pair: updateNmax failed");
}
void Cuda_Pair_GenerateXType(cuda_shared_data* sdata)
{
MYDBG(printf(" # CUDA: GenerateXType ... start %i %i %i %p %p %p %p\n",sdata->atom.nlocal,sdata->atom.nall,sdata->atom.nmax,sdata->atom.x.dev_data,sdata->atom.x_type.dev_data,sdata->atom.xhold.dev_data,sdata->atom.type.dev_data); )
if(sdata->atom.update_nmax)
Cuda_Pair_UpdateNmax(sdata);
if(sdata->atom.update_nlocal)
{
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
}
MYDBG(printf(" # CUDA: GenerateXType ... getgrid\n"); fflush(stdout); )
int3 layout=getgrid(sdata->atom.nall);
dim3 threads(layout.z, 1, 1);
dim3 grid(layout.x, layout.y, 1);
MYDBG(printf(" # CUDA: GenerateXType ... kernel start test\n"); fflush(stdout);)
Pair_GenerateXType_Kernel<<<grid, threads,0>>>();
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair GenerateXType: Kernel failed");
MYDBG(printf(" # CUDA: GenerateXType ... end\n"); fflush(stdout); )
}
void Cuda_Pair_RevertXType(cuda_shared_data* sdata)
{
MYDBG(printf(" # CUDA: RevertXType ... start\n"); )
if(sdata->atom.update_nmax)
Cuda_Pair_UpdateNmax(sdata);
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
int3 layout=getgrid(sdata->atom.nall);
dim3 threads(layout.z, 1, 1);
dim3 grid(layout.x, layout.y, 1);
Pair_RevertXType_Kernel<<<grid, threads,0>>>();
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair GenerateXType: Kernel failed");
MYDBG(printf(" # CUDA: RevertXType ... end\n"); )
}
void Cuda_Pair_GenerateVRadius(cuda_shared_data* sdata)
{
MYDBG(printf(" # CUDA: GenerateVRadius ... start %i %i %i %p %p %p %p\n",sdata->atom.nlocal,sdata->atom.nall,sdata->atom.nmax,sdata->atom.x.dev_data,sdata->atom.x_type.dev_data,sdata->atom.xhold.dev_data,sdata->atom.type.dev_data); )
if(sdata->atom.update_nmax)
Cuda_Pair_UpdateNmax(sdata);
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
MYDBG(printf(" # CUDA: GenerateVRadius ... getgrid\n"); fflush(stdout); )
int3 layout=getgrid(sdata->atom.nall);
dim3 threads(layout.z, 1, 1);
dim3 grid(layout.x, layout.y, 1);
MYDBG(printf(" # CUDA: GenerateVRadius ... kernel start test\n"); fflush(stdout);)
Pair_GenerateVRadius_Kernel<<<grid, threads,0>>>();
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair GenerateVRadius: Kernel failed");
MYDBG(printf(" # CUDA: GenerateVRadius ... end\n"); fflush(stdout); )
}
void Cuda_Pair_GenerateOmegaRmass(cuda_shared_data* sdata)
{
MYDBG(printf(" # CUDA: GenerateOmegaRmass ... start %i %i %i %p %p %p %p\n",sdata->atom.nlocal,sdata->atom.nall,sdata->atom.nmax,sdata->atom.x.dev_data,sdata->atom.x_type.dev_data,sdata->atom.xhold.dev_data,sdata->atom.type.dev_data); )
if(sdata->atom.update_nmax)
Cuda_Pair_UpdateNmax(sdata);
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
MYDBG(printf(" # CUDA: GenerateOmegaRmass ... getgrid\n"); fflush(stdout); )
int3 layout=getgrid(sdata->atom.nall);
dim3 threads(layout.z, 1, 1);
dim3 grid(layout.x, layout.y, 1);
MYDBG(printf(" # CUDA: GenerateOmegaRmass ... kernel start test\n"); fflush(stdout);)
Pair_GenerateOmegaRmass_Kernel<<<grid, threads,0>>>();
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair GenerateOmegaRmass: Kernel failed");
MYDBG(printf(" # CUDA: GenerateOmegaRmass ... end\n"); fflush(stdout); )
}
void Cuda_Pair_BuildXHold(cuda_shared_data* sdata)
{
if(sdata->atom.update_nmax)
Cuda_Pair_UpdateNmax(sdata);
cudaMemcpyToSymbol(MY_CONST(nlocal) , & sdata->atom.nlocal , sizeof(int) );
cudaMemcpyToSymbol(MY_CONST(nall) , & sdata->atom.nall , sizeof(int) );
int3 layout=getgrid(sdata->atom.nall);
dim3 threads(layout.z, 1, 1);
dim3 grid(layout.x, layout.y, 1);
Pair_BuildXHold_Kernel<<<grid, threads,0>>>();
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair GenerateXType: Kernel failed");
}
void Cuda_Pair_CollectForces(cuda_shared_data* sdata,int eflag, int vflag)
{
cudaThreadSynchronize();
clock_gettime(CLOCK_REALTIME,&endpairtime);
sdata->cuda_timings.pair_kernel+=
endpairtime.tv_sec-startpairtime.tv_sec+1.0*(endpairtime.tv_nsec-startpairtime.tv_nsec)/1000000000;
CUT_CHECK_ERROR("Cuda_Pair: Kernel execution failed");
dim3 threads;
dim3 grid;
if(eflag||vflag)
{
int n=sdata->pair.lastgridsize;
grid.x=sdata->pair.n_energy_virial;
grid.y=1;
threads.x=128;
//printf("A grid.x: %i\n",grid.x);
MY_AP(PairVirialCompute_reduce)<<<grid,threads,threads.x*sizeof(ENERGY_FLOAT)>>>(n);
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair_CollectForces: virial compute Kernel execution failed");
}
int3 layout=getgrid(sdata->atom.nlocal);
threads.x = layout.z;
grid.x = layout.x;
grid.y = layout.y;
Pair_CollectForces_Kernel<<<grid, threads,0>>>(sdata->pair.n_energy_virial,sdata->pair.lastgridsize);
cudaThreadSynchronize();
CUT_CHECK_ERROR("Cuda_Pair_CollectForces: Force Summation Kernel execution failed");
}
Reference in New Issue View Git Blame Copy Permalink