polygraphy.util.cuda Namespace Reference

Classes
class	Cuda
class	DeviceBuffer
class	Stream

Functions
def	wrapper ()
def	try_get_stream_handle (stream)

Variables
	G_CUDA = None

Function Documentation

wrapper()

def polygraphy.util.cuda.wrapper

(

)

Here is the caller graph for this function:

try_get_stream_handle()

def polygraphy.util.cuda.try_get_stream_handle

(

stream

)

Here is the caller graph for this function:

Variable Documentation

G_CUDA

polygraphy.util.cuda.G_CUDA = None