Class Context

extensions/include/cuMat/src/Context.h:132–446

* \brief Stores the cuda context of the current thread. * cuMat uses one cuda stream per thread, and also potentially * different devices. * */

 * 
 */
class Context
{
private:
	cudaStream_t stream_;
	int device_ = 0;

#if CUMAT_CONTEXT_DEBUG_MEMORY==1
	int allocationsHost_ = 0;
	int allocationsDevice_ = 0;
#endif

	CUMAT_DISALLOW_COPY_AND_ASSIGN(Context);

public:
	Context(int device = 0)
		: device_(device)
		, stream_(nullptr)
	{
#if 0
		//stream creation must be synchronized
		//TODO: seems to work without
		static std::mutex mutex;
		std::lock_guard<std::mutex> lock(mutex);
#endif

#if CUMAT_SINGLE_THREAD_CONTEXT==0
		//each context has its own stream
		CUMAT_SAFE_CALL(cudaStreamCreateWithFlags(&stream_, cudaStreamNonBlocking));
#endif
		//for a global context, use the global stream

		//TODO: init BLAS context and so on

		CUMAT_LOG_DEBUG("Context initialized for thread 0x" << std::hex << std::this_thread::get_id() << ", stream: 0x" << stream_);
	}

	~Context()
	{
		if (stream_ != nullptr)
		{
			cudaStreamDestroy(stream_);
			stream_ = nullptr;
		}
		CUMAT_LOG_DEBUG("Context deleted for thread 0x" << std::hex << std::this_thread::get_id());
#if CUMAT_CONTEXT_DEBUG_MEMORY==1
		CUMAT_ASSERT(allocationsHost_ == 0 && "some host memory was not released");
		CUMAT_ASSERT(allocationsDevice_ == 0 && "some device memory was not released");
#endif
	}

	/**
	 * \brief Explicitly frees the context's resources.
	 * Warning: use with care! All future calls on this context will fail
	 * and the context can't be created again.
	 * Use only if you shut down the CUDA driver before the normal program
	 * termination.
	 */
	void destroy()