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

Go to the documentation of this file.

#ifndef LAMA_CUDA_STREAM_SYNC_TOKEN_HPP_
#define LAMA_CUDA_STREAM_SYNC_TOKEN_HPP_

// boost
#include <boost/shared_ptr.hpp>

#include <list>
#include <memory>

#include <cuda.h>
#if __GNUC__ >= 4 &&  __GNUC_MINOR__ > 6
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif //__GNUC__ >= 4 &&  __GNUC_MINOR__ > 6
#include <cuda_runtime.h> /* no diagnostic for this one */
#if __GNUC__ >= 4 &&  __GNUC_MINOR__ > 6
#pragma GCC diagnostic pop
#endif //__GNUC__ >= 4&&  __GNUC_MINOR__ > 6

#include <lama/config.hpp>
#include <lama/SyncToken.hpp>

namespace lama
{

class CUDAContext;

typedef boost::shared_ptr<const CUDAContext> CUDAContextPtr;

class LAMA_DLL_IMPORTEXPORT CUDAStreamSyncToken: public lama::SyncToken

{

public:

    friend class CUDAStreamSyncTokenPtr;

    CUDAStreamSyncToken( CUDAContextPtr context, CUstream stream );

    virtual ~CUDAStreamSyncToken();

    virtual void wait();

    virtual bool probe() const;

    cudaStream_t getCUDAStream() const;

    void createTimingEvent( CUevent& event ) const;

    void createEvent( CUevent& event ) const;

    void getTime( float* time, CUevent& startEvent, CUevent& stopEvent ) const ;

    bool probeEvent( const CUevent& stopEvent ) const;

    void recordEvent( const CUevent event );

    bool queryEvent( const CUevent event ) const;

    void synchronizeEvent( const CUevent event ) const;

private:

    CUDAContextPtr mCUDAContext;  // needed for synchronization

    const CUstream mStream;
    CUevent        mEvent;
};

class LAMA_DLL_IMPORTEXPORT CUDAStreamSyncTokenPtr
{
public:

    CUDAStreamSyncTokenPtr() throw();

    CUDAStreamSyncTokenPtr( CUDAStreamSyncTokenPtr& other ) throw();

    CUDAStreamSyncTokenPtr( std::auto_ptr<CUDAStreamSyncToken> other ) throw();

    CUDAStreamSyncTokenPtr( CUDAStreamSyncToken* pointer ) throw();

    ~CUDAStreamSyncTokenPtr();

    CUDAStreamSyncTokenPtr& operator=( std::auto_ptr<CUDAStreamSyncToken> other ) throw();

    CUDAStreamSyncTokenPtr& operator=( CUDAStreamSyncTokenPtr& other ) throw();

    operator std::auto_ptr<SyncToken>& ();

    CUDAStreamSyncToken* operator->() const throw();

    CUDAStreamSyncToken& operator*() const throw();

    CUDAStreamSyncToken* release() throw();

private:
    std::auto_ptr<CUDAStreamSyncToken> mCUDAStreamSyncToken;
    std::auto_ptr<SyncToken> mSyncToken;
};

}

#endif // LAMA_CUDA_STREAM_SYNC_TOKEN_HPP_