forked from lijiext/lammps
81 lines
2.3 KiB
Plaintext
81 lines
2.3 KiB
Plaintext
/***************************************************************************
|
|
pair_gpu_nbor.cu
|
|
-------------------
|
|
W. Michael Brown
|
|
|
|
Neighbor memory operations for LAMMPS GPU Library
|
|
|
|
__________________________________________________________________________
|
|
This file is part of the LAMMPS GPU Library
|
|
__________________________________________________________________________
|
|
|
|
begin : Tue Aug 4 2009
|
|
copyright : (C) 2009 by W. Michael Brown
|
|
email : wmbrown@sandia.gov
|
|
***************************************************************************/
|
|
|
|
/* -----------------------------------------------------------------------
|
|
Copyright (2009) 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.
|
|
----------------------------------------------------------------------- */
|
|
|
|
#include "pair_gpu_nbor.h"
|
|
|
|
int PairGPUNbor::bytes_per_atom(const int max_nbors) const {
|
|
if (_use_packing)
|
|
return (max_nbors*2+4)*sizeof(int);
|
|
else
|
|
return (max_nbors+3)*sizeof(int);
|
|
}
|
|
|
|
void PairGPUNbor::init(const int ij_size, const int max_atoms,
|
|
const int max_nbors) {
|
|
if (allocated)
|
|
clear();
|
|
|
|
// Initialize timers for the selected GPU
|
|
time_nbor.init();
|
|
|
|
if (_use_packing)
|
|
dev_nbor.safe_alloc(max_nbors+4,max_atoms);
|
|
else
|
|
dev_nbor.safe_alloc(3,max_atoms);
|
|
|
|
ij.safe_alloc(max_nbors*max_atoms);
|
|
host_ij.safe_alloc_w(ij_size);
|
|
|
|
allocated=true;
|
|
}
|
|
|
|
void PairGPUNbor::clear() {
|
|
if (!allocated)
|
|
return;
|
|
allocated=false;
|
|
|
|
ij.clear();
|
|
host_ij.clear();
|
|
dev_nbor.clear();
|
|
}
|
|
|
|
double PairGPUNbor::host_memory_usage() const {
|
|
return IJ_SIZE*sizeof(int)+sizeof(PairGPUNbor);
|
|
}
|
|
|
|
void PairGPUNbor::reset(const int inum, int *ilist, const int *numj,
|
|
cudaStream_t &s) {
|
|
ij_total=0;
|
|
|
|
dev_nbor.copy_from_host(ilist,inum);
|
|
int acc=0;
|
|
for (int i=0; i<inum; i++) {
|
|
host_ij[i]=numj[ilist[i]];
|
|
host_ij[i+inum]=acc;
|
|
acc+=numj[ilist[i]];
|
|
}
|
|
|
|
host_ij.copy_to_2Ddevice(dev_nbor.begin()+dev_nbor.row_size(),
|
|
dev_nbor.row_size(),2,inum, s);
|
|
}
|