energyMinimizerAdam.cu File Reference

#include "energyMinimizerAdam.cuh"

Include dependency graph for energyMinimizerAdam.cu:

Functions
__global__ void	gpu_adam_step_kernel (dVec *force, dVec *biasedMomentum, dVec *biasedMomentum2, dVec *correctedMomentum, dVec *correctedMomentum2, dVec *displacement, scalar deltaT, scalar beta1, scalar beta2, scalar beta1t, scalar beta2t, int N)
bool	gpu_adam_step (dVec *force, dVec *biasedMomentum, dVec *biasedMomentum2, dVec *correctedMomentum, dVec *correctedMomentum2, dVec *displacement, scalar deltaT, scalar beta1, scalar beta2, scalar beta1t, scalar beta2t, int N, int blockSize)
	run an adam minimization step More...