CudaBSplineGeoTransformer< T > Class Template Reference

#include <cuda_bspline_geo_transformer.h>

Inheritance diagram for CudaBSplineGeoTransformer< T >:

Collaboration diagram for CudaBSplineGeoTransformer< T >:

Public Member Functions
	CudaBSplineGeoTransformer ()
virtual	~CudaBSplineGeoTransformer ()
void	setSrc (const T *data) override
const T *	getSrc () const override
T *	getDest () const override
void	copySrcToDest () override
T *	interpolate (const std::vector< float > &matrices) override
void	sum (T *dest, const std::vector< float > &weights, size_t firstN, T norm) override
Public Member Functions inherited from BSplineGeoTransformer< T >
	BSplineGeoTransformer ()
	BSplineGeoTransformer (const BSplineGeoTransformer &)=delete
	BSplineGeoTransformer (const BSplineGeoTransformer &&)=delete
virtual	~BSplineGeoTransformer ()
BSplineGeoTransformer &	operator= (const BSplineGeoTransformer &)=delete
BSplineGeoTransformer &	operator= (const BSplineGeoTransformer &&)=delete
Public Member Functions inherited from AGeoTransformer< BSplineTransformSettings< T >, T >
	AGeoTransformer ()
virtual	~AGeoTransformer ()=default
void	init (const BSplineTransformSettings< T > &s, bool reuse)
const BSplineTransformSettings< T > &	getSettings () const

Additional Inherited Members
Protected Member Functions inherited from BSplineGeoTransformer< T >
virtual bool	canBeReused (const BSplineTransformSettings< T > &s) const override
Protected Member Functions inherited from AGeoTransformer< BSplineTransformSettings< T >, T >
constexpr bool	isInitialized () const
constexpr bool	isSrcSet () const
void	setIsSrcSet (bool status)

Detailed Description

template<typename T>
class CudaBSplineGeoTransformer< T >

Definition at line 36 of file cuda_bspline_geo_transformer.h.

Constructor & Destructor Documentation

CudaBSplineGeoTransformer()

template<typename T>

CudaBSplineGeoTransformer< T >::CudaBSplineGeoTransformer ( )

inline

Definition at line 38 of file cuda_bspline_geo_transformer.h.

                                {
        setDefault();
    }

~CudaBSplineGeoTransformer()

template<typename T>

virtual CudaBSplineGeoTransformer< T >::~CudaBSplineGeoTransformer ( )

inlinevirtual

Definition at line 42 of file cuda_bspline_geo_transformer.h.

                                         {
        release();
    }

Member Function Documentation

copySrcToDest()

template<typename T >

void CudaBSplineGeoTransformer< T >::copySrcToDest ( )

overridevirtual

Reimplemented from BSplineGeoTransformer< T >.

Definition at line 78 of file cuda_bspline_geo_transformer.cpp.

                                                 {
    bool isReady = this->isInitialized()
            && this->isSrcSet();
    if ( ! isReady) {
        REPORT_ERROR(ERR_LOGIC_ERROR, "Instance is either not initialized or the 'src' has not been set.");
    }
    gpuErrchk(cudaMemcpy(
        m_d_dest,
        m_d_src,
        this->getSettings().dims.size() * sizeof(T),
        cudaMemcpyDeviceToDevice)); // FIXME DS use proper stream
}

getDest()

template<typename T>

T* CudaBSplineGeoTransformer< T >::getDest ( ) const

inlineoverridevirtual

Reimplemented from BSplineGeoTransformer< T >.

Definition at line 52 of file cuda_bspline_geo_transformer.h.

                                {
        return m_d_dest;
    }

getSrc()

template<typename T>

const T* CudaBSplineGeoTransformer< T >::getSrc ( ) const

inlineoverridevirtual

Reimplemented from BSplineGeoTransformer< T >.

Definition at line 48 of file cuda_bspline_geo_transformer.h.

                                     {
        return m_d_src;
    }

interpolate()

template<typename T >

T * CudaBSplineGeoTransformer< T >::interpolate ( const std::vector< float > &  matrices )

overridevirtual

Reimplemented from BSplineGeoTransformer< T >.

Definition at line 121 of file cuda_bspline_geo_transformer.cpp.

                                                                             {
    const auto& dims = this->getSettings().dims;
    size_t bytes = dims.n() * 9 * sizeof(float);
    auto stream = *(cudaStream_t*)m_stream->stream();
    gpuErrchk(cudaMemcpyAsync(
            m_d_matrices,
            matrices.data(),
            bytes,
            cudaMemcpyHostToDevice, stream));
    dim3 dimBlock(64, 1, 1);
    dim3 dimGrid(
        std::ceil(dims.x() / (float)dimBlock.x),
        std::ceil(dims.y() / (float)dimBlock.y),
        dims.n());

    interpolateKernel<<<dimGrid, dimBlock, 0, stream>>> (
                    m_d_src, m_d_dest, m_d_matrices,
                    dims.x(), dims.y());
    m_stream->synch();

    return m_d_dest;
}

setSrc()

template<typename T >

void CudaBSplineGeoTransformer< T >::setSrc ( const T *  data )

overridevirtual

Reimplemented from BSplineGeoTransformer< T >.

Definition at line 32 of file cuda_bspline_geo_transformer.cpp.

                                                         {
    const auto& s = this->getSettings();
    gpuErrchk(cudaMemcpy(
        m_d_src,
        h_data,
        this->getSettings().dims.size() * sizeof(T),
        cudaMemcpyHostToDevice));
    this->setIsSrcSet(true);
}

sum()

template<typename T >

void CudaBSplineGeoTransformer< T >::sum ( T *  dest, const std::vector< float > &  weights, size_t  firstN, T  norm  )

overridevirtual

Reimplemented from BSplineGeoTransformer< T >.

Definition at line 145 of file cuda_bspline_geo_transformer.cpp.

                {
    const auto& dims = this->getSettings().dims.copyForN(firstN);
    assert(weights.size() == dims.n());
    assert(0 < dims.n());
    dim3 dimBlock(32, 32, 1);
    dim3 dimGrid(
        std::ceil(dims.x() / (float)dimBlock.x),
        std::ceil(dims.y() / (float)dimBlock.y),
        1);
    auto stream = *(cudaStream_t*)m_stream->stream();

    // copy weights
    T *d_weights;
    size_t bytesWeight = weights.size() * sizeof(float);
    gpuErrchk(cudaMalloc(&d_weights, bytesWeight));
    gpuErrchk(cudaMemcpyAsync(
            d_weights,
            weights.data(),
            bytesWeight,
            cudaMemcpyHostToDevice, stream));

    sumKernel<<<dimGrid, dimBlock, 0, stream>>> (
        m_d_dest, m_d_src, d_weights,// FIXME DS this will damage src data
        dims.x(), dims.y(), dims.n(), norm);

    size_t bytes = dims.sizeSingle() * sizeof(T);
    gpuErrchk(cudaMemcpyAsync(
        dest,
        m_d_src,
        bytes,
        cudaMemcpyDeviceToHost, stream));

    gpuErrchk(cudaFree(d_weights));
    m_stream->synch();

}

---

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

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