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

HandleState

enum mshadow::Stream< gpu >::HandleState

handle state

Enumerator
NoHandle
OwnHandle

Constructor & Destructor Documentation

Stream()

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

inline

Member Function Documentation

CheckIdle()

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

CreateBlasHandle()

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

inline

Destory original blas handle and create a new one.

CreateDnnHandle()

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

inline

CreateSolverHandle()

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

inline

DestroyBlasHandle()

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

inline

Destory cublas handle if own it.

DestroyDnnHandle()

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

inline

DestroySolverHandle()

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

inline

GetBlasHandle()

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

inlinestatic

return actual cublasHandle

Parameters

pointer

to GPU stream

GetSolverHandle()

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

inlinestatic

GetStream()

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

Wait()

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

inline

wait for all the computation associated with this stream to complete

Member Data Documentation

blas_handle_

cublasHandle_t mshadow::Stream< gpu >::blas_handle_

cublas handle

blas_handle_ownership_

HandleState mshadow::Stream< gpu >::blas_handle_ownership_

cudnn handle

cublas handle ownership

dev_id

int mshadow::Stream< gpu >::dev_id

dev id

dnn_handle_ownership_

HandleState mshadow::Stream< gpu >::dnn_handle_ownership_

cudnn handle ownership

prop

cudaDeviceProp mshadow::Stream< gpu >::prop

cudaDeviceProp

solver_handle_

cusolverDnHandle_t mshadow::Stream< gpu >::solver_handle_

cusolver handle

solver_handle_ownership_

HandleState mshadow::Stream< gpu >::solver_handle_ownership_

cusolver handle ownership

stream_

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