mshadow::Stream< gpu > Struct Template Reference

#include <stream_gpu-inl.h>

Collaboration diagram for mshadow::Stream< gpu >:

Public Types
enum	HandleState { NoHandle = 0, OwnHandle = 1 }
	handle state More...

Public Member Functions
	Stream (void)
void	Wait (void)
	wait for all the computation associated with this stream to complete More...
bool	CheckIdle (void)
	query whether the the stream is idle More...
void	DestroyBlasHandle ()
	Destory cublas handle if own it. More...
void	CreateBlasHandle ()
	Destory original blas handle and create a new one. More...
void	DestroySolverHandle ()
void	CreateSolverHandle ()
void	DestroyDnnHandle ()
void	CreateDnnHandle ()

Static Public Member Functions
static cudaStream_t	GetStream (Stream< gpu > *stream)
	returns actual cudaStream_t given an input GPU stream pointer More...
static cublasHandle_t	GetBlasHandle (Stream< gpu > *stream)
	return actual cublasHandle More...
static cusolverDnHandle_t	GetSolverHandle (Stream< gpu > *stream)

Public Attributes
cudaStream_t	stream_
	cudaStream More...
cublasHandle_t	blas_handle_
	cublas handle More...
cusolverDnHandle_t	solver_handle_
	cusolver handle More...
HandleState	blas_handle_ownership_
	cudnn handle More...
HandleState	solver_handle_ownership_
	cusolver handle ownership More...
HandleState	dnn_handle_ownership_
	cudnn handle ownership More...
cudaDeviceProp	prop
	cudaDeviceProp More...
int	dev_id
	dev id More...

Member Enumeration Documentation

enum mshadow::Stream< gpu >::HandleState

handle state

Enumerator
NoHandle
OwnHandle

Constructor & Destructor Documentation

mshadow::Stream< gpu >::Stream ( void  )

inline

Member Function Documentation

bool mshadow::Stream< gpu >::CheckIdle ( void  )

inline

query whether the the stream is idle

Returns

true if the stream is idle and all the job have been completed

void mshadow::Stream< gpu >::CreateBlasHandle ( )

inline

Destory original blas handle and create a new one.

void mshadow::Stream< gpu >::CreateDnnHandle ( )

inline

void mshadow::Stream< gpu >::CreateSolverHandle ( )

inline

void mshadow::Stream< gpu >::DestroyBlasHandle ( )

inline

Destory cublas handle if own it.

void mshadow::Stream< gpu >::DestroyDnnHandle ( )

inline

void mshadow::Stream< gpu >::DestroySolverHandle ( )

inline

static cublasHandle_t mshadow::Stream< gpu >::GetBlasHandle ( Stream< gpu > *  stream )

inlinestatic

return actual cublasHandle

Parameters

pointer

to GPU stream

static cusolverDnHandle_t mshadow::Stream< gpu >::GetSolverHandle ( Stream< gpu > *  stream )

inlinestatic

static cudaStream_t mshadow::Stream< gpu >::GetStream ( Stream< gpu > *  stream )

inlinestatic

returns actual cudaStream_t given an input GPU stream pointer

Parameters

stream

pointer to GPU stream

void mshadow::Stream< gpu >::Wait ( void  )

inline

wait for all the computation associated with this stream to complete

Member Data Documentation

cublasHandle_t mshadow::Stream< gpu >::blas_handle_

cublas handle

HandleState mshadow::Stream< gpu >::blas_handle_ownership_

cudnn handle

cublas handle ownership

int mshadow::Stream< gpu >::dev_id

dev id

HandleState mshadow::Stream< gpu >::dnn_handle_ownership_

cudnn handle ownership

cudaDeviceProp mshadow::Stream< gpu >::prop

cudaDeviceProp

cusolverDnHandle_t mshadow::Stream< gpu >::solver_handle_

cusolver handle

HandleState mshadow::Stream< gpu >::solver_handle_ownership_

cusolver handle ownership

cudaStream_t mshadow::Stream< gpu >::stream_

cudaStream

---

The documentation for this struct was generated from the following file:

• /work/mxnet/3rdparty/mshadow/mshadow/stream_gpu-inl.h