CudaPmeSolverUtil.C File Reference

#include <stdio.h>
#include <algorithm>
#include <cstdlib>
#include <cuda_runtime.h>
#include "HipDefines.h"
#include "ComputeNonbondedUtil.h"
#include "ComputePmeCUDAMgr.h"
#include "CudaPmeSolver.h"
#include "CudaPmeSolverUtil.h"
#include "Node.h"
#include "PatchData.h"
#include "NamdEventsProfiling.h"
#include "TestArray.h"
#include "DeviceCUDA.h"

Go to the source code of this file.

Functions
void	CcdCallBacksReset (void *ignored, double curWallTime)
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)

Variables
__thread DeviceCUDA *	deviceCUDA

Function Documentation

CcdCallBacksReset()

void CcdCallBacksReset	(	void *	ignored,
		double	curWallTime
	)

Referenced by CudaPmeKSpaceCompute::energyAndVirialSetCallback(), and CudaPmeRealSpaceCompute::gatherForceSetCallback().

writeComplexToDisk()

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

Definition at line 30 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 42 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 49 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;
}

Variable Documentation

deviceCUDA

__thread DeviceCUDA* deviceCUDA

Definition at line 23 of file DeviceCUDA.C.

Referenced by CudaPmeOneDevice::checkPatchLevelSimParamCompatibility().