CudaRandom.h

#ifndef PSCF_CUDA_RANDOM_H
#define PSCF_CUDA_RANDOM_H
 
#include "DeviceArray.h"
 
#include <curand.h>
 
/*
* PSCF Package - Polymer Self-Consistent Field 
*
* Copyright 2016 - 2023, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
namespace Pscf {
 
   class CudaRandom 
   {
 
   public:
 
      CudaRandom();
 
      virtual ~CudaRandom();
   
      void setSeed(unsigned long long seed);
 
      void uniform(DeviceArray<float>& data);
 
      void uniform(DeviceArray<double>& data);
   
      void normal(DeviceArray<float>& data, float stddev, float mean = 0.0);
 
      void normal(DeviceArray<double>& data, double stddev, double mean = 0.0);
   
      long seed();
   
   private:
 
      curandGenerator_t gen_;
 
      unsigned long long seed_;
 
      bool isInitialized_;
 
      void errorCheck(curandStatus_t const & error);
   
   };
 
   /* 
   * Returns value of random seed (private member variable idum)
   */
   inline long CudaRandom::seed() 
   {  return seed_; }
 
} 
#endif