TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
polygraphy.util.cuda.Cuda Class Reference
Inheritance diagram for polygraphy.util.cuda.Cuda:
Collaboration diagram for polygraphy.util.cuda.Cuda:

Classes

class  MemcpyKind
 

Public Member Functions

def __init__ (self)
 
def check (self, status)
 
def create_stream (self)
 
def stream_synchronize (self, stream)
 
def destroy_stream (self, stream)
 
def malloc (self, nbytes)
 
def free (self, ptr)
 
def htod (self, dst, src, nbytes, stream=None)
 
def dtoh (self, dst, src, nbytes, stream=None)
 

Public Attributes

 handle
 

Constructor & Destructor Documentation

◆ __init__()

def polygraphy.util.cuda.Cuda.__init__ (   self)

Member Function Documentation

◆ check()

def polygraphy.util.cuda.Cuda.check (   self,
  status 
)
Here is the caller graph for this function:

◆ create_stream()

def polygraphy.util.cuda.Cuda.create_stream (   self)
Here is the call graph for this function:

◆ stream_synchronize()

def polygraphy.util.cuda.Cuda.stream_synchronize (   self,
  stream 
)
Here is the call graph for this function:

◆ destroy_stream()

def polygraphy.util.cuda.Cuda.destroy_stream (   self,
  stream 
)
Here is the call graph for this function:

◆ malloc()

def polygraphy.util.cuda.Cuda.malloc (   self,
  nbytes 
)
Here is the call graph for this function:

◆ free()

def polygraphy.util.cuda.Cuda.free (   self,
  ptr 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ htod()

def polygraphy.util.cuda.Cuda.htod (   self,
  dst,
  src,
  nbytes,
  stream = None 
)
Here is the call graph for this function:

◆ dtoh()

def polygraphy.util.cuda.Cuda.dtoh (   self,
  dst,
  src,
  nbytes,
  stream = None 
)
Here is the call graph for this function:

Member Data Documentation

◆ handle

polygraphy.util.cuda.Cuda.handle

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