ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

do GPU device memory accounting with doubles

Browse Source 
using (signed) int will overflow at 2GB,
switching to unsigned has risk of hiding
overflows and using long long is not as
portable as double precisiong floating point.
This commit is contained in:
Axel Kohlmeyer
2010-11-11 12:31:36 -05:00
parent 92444042fc
commit 92cd2a69bf
7 changed files with 22 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/gpu/pair_gpu_device.cpp
View File

@ -166,7 +166,7 @@ void PairGPUDeviceT::init_message(FILE *screen, const char *name,
template <class numtyp, class acctyp>
void PairGPUDeviceT::output_times(UCL_Timer &time_pair, const double avg_split,
const int max_bytes, FILE *screen) {
const double max_bytes, FILE *screen) {
double single[5], times[5];
single[0]=atom.transfer_time();
@ -177,10 +177,10 @@ void PairGPUDeviceT::output_times(UCL_Timer &time_pair, const double avg_split,
MPI_Reduce(single,times,5,MPI_DOUBLE,MPI_SUM,0,MPI_COMM_WORLD);
int my_max_bytes=max_bytes;
int mpi_max_bytes;
MPI_Reduce(&my_max_bytes,&mpi_max_bytes,1,MPI_INT,MPI_MAX,0,MPI_COMM_WORLD);
double max_mb=mpi_max_bytes/(1024*1024);
double my_max_bytes=max_bytes;
double mpi_max_bytes;
MPI_Reduce(&my_max_bytes,&mpi_max_bytes,1,MPI_DOUBLE,MPI_MAX,0,MPI_COMM_WORLD);
double max_mb=mpi_max_bytes/(1024.0*1024.0);
if (world_me()==0)
if (screen && times[3]>0.0) {