Gpu::CDF< T > Struct Template Reference

#include <cuda_cdf.h>

Collaboration diagram for Gpu::CDF< T >:

Public Member Functions
	CDF (size_t volume_size, T multConst=1.0, T probStep=0.005)
	~CDF ()
void	calculateCDF (const T *d_filtered1, const T *d_filtered2)
void	calculateCDF (const T *d_S)
void	_calculateDifference (const T *__restrict__ d_filtered1, const T *__restrict__ d_filtered2)
void	_calculateSquare (const T *__restrict__ d_S)
void	_updateProbabilities ()

Public Attributes
T *	d_V
T *	d_x
T *	d_probXLessThanx
size_t	volume_size
T	probStep
T	multConst
T	Nsteps

Static Public Attributes
static constexpr size_t	type_size = sizeof(T)

Detailed Description

template<typename T>
struct Gpu::CDF< T >

Gpu version of Cumulative density function. This function computes a table with the cumulative density function

Definition at line 33 of file cuda_cdf.h.

Constructor & Destructor Documentation

CDF()

template<typename T >

CDF::CDF	(	size_t	volume_size,
		T	multConst = 1.0,
		T	probStep = 0.005
	)

Definition at line 41 of file cuda_cdf.cpp.

: volume_size(volume_size)
, multConst(multConst)
, probStep(probStep)
, Nsteps(round(1.0/probStep)) {
    gpuErrchk( cudaMalloc((void**)&d_V, volume_size * type_size) );
    gpuErrchk( cudaMalloc((void**)&d_x, Nsteps * type_size) );
    gpuErrchk( cudaMalloc((void**)&d_probXLessThanx, Nsteps * type_size) );
}

~CDF()

template<typename T >

CDF::~CDF

(

)

Definition at line 52 of file cuda_cdf.cpp.

             {
    gpuErrchk( cudaFree(d_V) );
    gpuErrchk( cudaFree(d_x) );
    gpuErrchk( cudaFree(d_probXLessThanx) );
}

Member Function Documentation

_calculateDifference()

template<typename T >

void CDF::_calculateDifference	(	const T *__restrict__	d_filtered1,
		const T *__restrict__	d_filtered2
	)

Definition at line 73 of file cuda_cdf.cpp.

                                                                                                      {
    T* __restrict__ d_V = this->d_V;
    const T multConst = this->multConst;

    auto compute_diff = [=] __device__ (int index) {
        T diff = d_filtered1[index] - d_filtered2[index];
        d_V[index] = multConst * diff * diff;
    };

    thrust::for_each_n(thrust::device, thrust::counting_iterator<int>(0), volume_size, compute_diff);
}

_calculateSquare()

template<typename T >

void CDF::_calculateSquare

(

const T *__restrict__

d_S

)

Definition at line 86 of file cuda_cdf.cpp.

                                                       {
    T* __restrict__ d_V = this->d_V;

    auto k = [=] __host__ __device__ (int index) {
        T val = d_S[index];
        d_V[index] = val * val;
    };

    thrust::for_each_n(thrust::device, thrust::counting_iterator<int>(0), volume_size, k);
}

_updateProbabilities()

template<typename T >

void CDF::_updateProbabilities

(

)

Definition at line 103 of file cuda_cdf.cpp.

                                  {
    T* __restrict__ d_V = this->d_V;
    T* __restrict__ kd_x = this->d_x;
    T* __restrict__ kd_probXLessThanx = this->d_probXLessThanx;
    const T kprobStep = this->probStep;
    const size_t k_volume_size = this->volume_size;

    auto k = [d_V, kd_x, kd_probXLessThanx, kprobStep, k_volume_size] __device__ (int index) {
        int i = 0;
        for (T p = kprobStep / 2; p < 1; p += kprobStep, i++) {
            int idx = static_cast<int>(round(p * k_volume_size));
            kd_probXLessThanx[i] = p;
            kd_x[i] = d_V[idx];
        }
    };

    thrust::for_each_n(thrust::device, thrust::counting_iterator<int>(0), 1, k);
}

calculateCDF() [1/2]

template<typename T >

void CDF::calculateCDF	(	const T *	d_filtered1,
		const T *	d_filtered2
	)

Definition at line 59 of file cuda_cdf.cpp.

                                                                    {
    _calculateDifference(d_filtered1, d_filtered2);
    sort();
    _updateProbabilities();
}

calculateCDF() [2/2]

template<typename T >

void CDF::calculateCDF

(

const T *

d_S

)

Definition at line 66 of file cuda_cdf.cpp.

                                      {
    _calculateSquare(d_S);
    sort();
    _updateProbabilities();
}

Member Data Documentation

d_probXLessThanx

template<typename T>

T* Gpu::CDF< T >::d_probXLessThanx

Definition at line 38 of file cuda_cdf.h.

d_V

template<typename T>

T* Gpu::CDF< T >::d_V

Definition at line 36 of file cuda_cdf.h.

d_x

template<typename T>

T* Gpu::CDF< T >::d_x

Definition at line 37 of file cuda_cdf.h.

multConst

template<typename T>

T Gpu::CDF< T >::multConst

Definition at line 42 of file cuda_cdf.h.

Nsteps

template<typename T>

T Gpu::CDF< T >::Nsteps

Definition at line 43 of file cuda_cdf.h.

probStep

template<typename T>

T Gpu::CDF< T >::probStep

Definition at line 41 of file cuda_cdf.h.

type_size

template<typename T>

constexpr size_t Gpu::CDF< T >::type_size = sizeof(T)

static

Definition at line 34 of file cuda_cdf.h.

volume_size

template<typename T>

size_t Gpu::CDF< T >::volume_size

Definition at line 40 of file cuda_cdf.h.

---

The documentation for this struct was generated from the following files:

• xmipp/libraries/reconstruction_cuda/cuda_cdf.h
• xmipp/libraries/reconstruction_cuda/cuda_cdf.cpp