nll_kernels.h

Go to the documentation of this file.

#ifndef __NLL_H__
#define __NLL_H__

#include <cuda.h>
#include <hemi/hemi.h>

#ifdef __CUDACC__
#include <curand_kernel.h>
#endif

#ifndef __HEMI_ARRAY_H__
#define __HEMI_ARRAY_H__
#include <hemi/array.h>
#endif

#ifdef __CUDACC__
typedef curandStateXORWOW RNGState;
#else
typedef int RNGState;
#endif

class TNtuple;

#ifdef __CUDACC__

__global__ void init_device_rngs(int nthreads, unsigned long long seed,
                                 curandState* state);
#endif


HEMI_KERNEL(pick_new_vector)(int nthreads, RNGState* rng, float sigma,
                             const float* current_vector,
                             float* proposed_vector);

HEMI_KERNEL(jump_decider)(RNGState* rng, double* nll_current,
                          const double* nll_proposed, float* v_current,
                          const float* v_proposed, unsigned ns, int* counter,
                          float* jump_buffer);

HEMI_KERNEL(nll_event_chunks)(const float* lut, const float* pars,
                              const size_t ne, const size_t ns,
                              double* sums);


HEMI_KERNEL(nll_event_reduce)(const size_t nthreads, const double* sums,
                              double* total_sum);


HEMI_KERNEL(nll_total)(const size_t ns, const float* pars,
                       const float* expectations,
                       const float* constraints,
                       const double* events_total,
                       double* nll);

#endif  // __NLL_H__

---