/home/brandes/workspace/LAMA/src/lama/cuda/CUDATracerHelper.hpp

Go to the documentation of this file.

#ifndef LAMA_CUDATRACERHELPER_HPP_
#define LAMA_CUDATRACERHELPER_HPP_

// for dll_import
#include <lama/config.hpp>

// others
#include <lama/tracing/LAMABaseTracer.hpp>
#include <lama/tracing/CUDATracerSyncToken.hpp>

#include <lama/cuda/CUDAStreamSyncToken.hpp>

template<typename Tracer>
class LAMA_DLL_IMPORTEXPORT CUDATracerHelper
{
public:
    CUDATracerHelper(const char* name, const char* file, int lno, lama::CUDAStreamSyncToken& cudaStreamSyncToken);
    virtual ~CUDATracerHelper();

private:
    CUDATracerSyncToken* mTracerSyncToken;
};

template<typename Tracer>
CUDATracerHelper<Tracer>::CUDATracerHelper(
    const char* name, const char* file, int lno,
    lama::CUDAStreamSyncToken& cudaStreamSyncToken)
    : mTracerSyncToken(0)
{
    std::auto_ptr<LAMABaseTracer> tracer( new Tracer( name, file, lno ) );

    std::auto_ptr<CUDATracerSyncToken> cudaTracerSyncToken( new CUDATracerSyncToken( tracer, cudaStreamSyncToken ) );
    mTracerSyncToken = cudaTracerSyncToken.get();
    std::auto_ptr<lama::SyncToken> tracerSyncToken( cudaTracerSyncToken );
    cudaStreamSyncToken.pushSyncToken( tracerSyncToken );
}

template<typename Tracer>
CUDATracerHelper<Tracer>::~CUDATracerHelper()
{
    mTracerSyncToken->recordStopEvent( );
}

#endif // LAMA_CUDATRACERHELPER_HPP_