CudaCorrelationComputer< T > Class Template Reference

#include <cuda_correlation_computer.h>

Inheritance diagram for CudaCorrelationComputer< T >:

Collaboration diagram for CudaCorrelationComputer< T >:

Public Member Functions
	CudaCorrelationComputer ()
	~CudaCorrelationComputer ()
void	loadReference (const T *ref) override
void	compute (T *others) override
Public Member Functions inherited from AMeritComputer< T >
	AMeritComputer ()
virtual	~AMeritComputer ()=default
void	init (const MeritSettings &s, bool reuse)
const MeritSettings &	getSettings () const
const std::vector< float > &	getFiguresOfMerit () const

Additional Inherited Members
Protected Member Functions inherited from AMeritComputer< T >
constexpr bool	isInitialized () const
constexpr bool	isRefLoaded () const
void	setIsRefLoaded (bool status)
std::vector< float > &	getFiguresOfMerit ()

Detailed Description

template<typename T>
class CudaCorrelationComputer< T >

Definition at line 37 of file cuda_correlation_computer.h.

Constructor & Destructor Documentation

CudaCorrelationComputer()

template<typename T>

CudaCorrelationComputer< T >::CudaCorrelationComputer ( )

inline

Definition at line 39 of file cuda_correlation_computer.h.

                              {
        setDefault();
    }

~CudaCorrelationComputer()

template<typename T>

CudaCorrelationComputer< T >::~CudaCorrelationComputer ( )

inline

Definition at line 43 of file cuda_correlation_computer.h.

                               {
        release();
    }

Member Function Documentation

compute()

template<typename T >

void CudaCorrelationComputer< T >::compute ( T *  others )

overridevirtual

Implements AMeritComputer< T >.

Definition at line 75 of file cuda_correlation_computer.cpp.

                                                  {
    bool isReady = this->isInitialized() && this->isRefLoaded();
    if ( ! isReady) {
        REPORT_ERROR(ERR_LOGIC_ERROR, "Not ready to execute. Call init() and load reference");
    }
    if ( ! m_stream->isGpuPointer(others)) {
        REPORT_ERROR(ERR_NOT_IMPLEMENTED, "Processing data from host is not yet supported");
    } else {
        m_d_others = others;
    }

    const auto &s = this->getSettings();
    this->getFiguresOfMerit().resize(s.otherDims.n());
    switch(s.type) {
        case MeritType::OneToN: {
            if (s.normalizeResult) {
                computeOneToN<true>();
            } else {
                computeOneToN<false>();
            }
            break;
        }
        default:
            REPORT_ERROR(ERR_NOT_IMPLEMENTED, "This case is not implemented");
    }
}

loadReference()

template<typename T >

void CudaCorrelationComputer< T >::loadReference ( const T *  ref )

overridevirtual

Implements AMeritComputer< T >.

Definition at line 103 of file cuda_correlation_computer.cpp.

                                                           {
    const auto &s = this->getSettings();
    this->setIsRefLoaded(nullptr != ref);

    if (m_stream->isGpuPointer(ref)) {
        REPORT_ERROR(ERR_NOT_IMPLEMENTED, "Only reference data on CPU are currently supported");
    } else {
        size_t bytes = s.refDims.size() * sizeof(T);
        bool hasToPin = ! m_stream->isMemoryPinned(ref);
        if (hasToPin) {
            m_stream->pinMemory(ref, bytes);
        }
        auto stream = *(cudaStream_t*)m_stream->stream();
        // copy data to GPU
        gpuErrchk(cudaMemcpyAsync(
            m_d_ref,
            ref,
            bytes,
            cudaMemcpyHostToDevice, stream));
        if (hasToPin) {
            m_stream->unpinMemory(ref);
        }
    }
    if (s.normalizeResult) {
        computeAvgStddevForRef();
    } else {
        // nothing to do, assume that stddev = 1 and avg = 0 for each signal
    }
}

---

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

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