CudaPmeSolverUtil.h File Reference

#include <cuda.h>
#include <cufft.h>
#include <stdio.h>
#include "PmeSolverUtil.h"
#include "CudaUtils.h"
#include "CudaPmeSolverUtilKernel.h"
#include "ReductionMgr.h"
#include "HipDefines.h"

Go to the source code of this file.

Classes
class	CudaFFTCompute
class	CudaPmeKSpaceCompute
class	CudaPmeRealSpaceCompute
class	CudaPmeTranspose
class	CudaPmeOneDevice
struct	CudaPmeOneDevice::EnergyVirial

Macros
#define	cufftCheck(stmt)

Functions
void	writeComplexToDisk (const float2 *d_data, const int size, const char *filename, cudaStream_t stream)
void	writeHostComplexToDisk (const float2 *h_data, const int size, const char *filename)
void	writeRealToDisk (const float *d_data, const int size, const char *filename, cudaStream_t stream)

Macro Definition Documentation

cufftCheck

#define cufftCheck

(

stmt

)

Value:

do {                                            \
  cufftResult err = stmt;                                               \
  if (err != CUFFT_SUCCESS) {                                           \
        char msg[128];  \
          sprintf(msg, "%s in file %s, function %s\n", #stmt,__FILE__,__FUNCTION__); \
          cudaDie(msg); \
  }                                                                     \
} while(0)

Definition at line 29 of file CudaPmeSolverUtil.h.

Referenced by CudaFFTCompute::backward(), CudaPmeOneDevice::compute(), CudaPmeOneDevice::CudaPmeOneDevice(), CudaFFTCompute::forward(), CudaFFTCompute::~CudaFFTCompute(), and CudaPmeOneDevice::~CudaPmeOneDevice().

Function Documentation

writeComplexToDisk()

void writeComplexToDisk	(	const float2 *	d_data,
		const int	size,
		const char *	filename,
		cudaStream_t	stream
	)

Definition at line 27 of file CudaPmeSolverUtil.C.

References cudaCheck.

                                                                                                         {
  fprintf(stderr, "writeComplexToDisk %d %s\n", size, filename);
  float2* h_data = new float2[size];
  copy_DtoH<float2>(d_data, h_data, size, stream);
  cudaCheck(cudaStreamSynchronize(stream));
  FILE *handle = fopen(filename, "w");
  for (int i=0;i < size;i++)
    fprintf(handle, "%f %f\n", h_data[i].x, h_data[i].y);
  fclose(handle);
  delete [] h_data;
}

writeHostComplexToDisk()

void writeHostComplexToDisk	(	const float2 *	h_data,
		const int	size,
		const char *	filename
	)

Definition at line 39 of file CudaPmeSolverUtil.C.

                                                                                        {
  FILE *handle = fopen(filename, "w");
  for (int i=0;i < size;i++)
    fprintf(handle, "%f %f\n", h_data[i].x, h_data[i].y);
  fclose(handle);
}

writeRealToDisk()

void writeRealToDisk	(	const float *	d_data,
		const int	size,
		const char *	filename,
		cudaStream_t	stream
	)

Definition at line 46 of file CudaPmeSolverUtil.C.

References cudaCheck.

                                                                                                     {
  fprintf(stderr, "writeRealToDisk %d %s\n", size, filename);
  float* h_data = new float[size];
  copy_DtoH<float>(d_data, h_data, size, stream);
  cudaCheck(cudaStreamSynchronize(stream));
  FILE *handle = fopen(filename, "w");
  for (int i=0;i < size;i++)
    fprintf(handle, "%f\n", h_data[i]);
  fclose(handle);
  delete [] h_data;
}