TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
Public Member Functions | Private Attributes | List of all members
nvinfer1::CUDADriverWrapper Class Reference
Collaboration diagram for nvinfer1::CUDADriverWrapper:
Collaboration graph
[legend]

Public Member Functions

 CUDADriverWrapper ()
 
 ~CUDADriverWrapper ()
 
CUresult cuGetErrorName (CUresult error, const char **pStr) const
 
CUresult cuFuncSetAttribute (CUfunction hfunc, CUfunction_attribute attrib, int value) const
 
CUresult cuLinkComplete (CUlinkState state, void **cubinOut, size_t *sizeOut) const
 
CUresult cuModuleUnload (CUmodule hmod) const
 
CUresult cuLinkDestroy (CUlinkState state) const
 
CUresult cuModuleLoadData (CUmodule *module, const void *image) const
 
CUresult cuLinkCreate (unsigned int numOptions, CUjit_option *options, void **optionValues, CUlinkState *stateOut) const
 
CUresult cuModuleGetFunction (CUfunction *hfunc, CUmodule hmod, const char *name) const
 
CUresult cuLinkAddFile (CUlinkState state, CUjitInputType type, const char *path, unsigned int numOptions, CUjit_option *options, void **optionValues) const
 
CUresult cuLinkAddData (CUlinkState state, CUjitInputType type, void *data, size_t size, const char *name, unsigned int numOptions, CUjit_option *options, void **optionValues) const
 
CUresult cuLaunchCooperativeKernel (CUfunction f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, unsigned int blockDimZ, unsigned int sharedMemBytes, CUstream hStream, void **kernelParams) const
 
CUresult cuLaunchKernel (CUfunction f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, unsigned int blockDimZ, unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, void **extra) const
 

Private Attributes

void * handle
 
CUresult(* _cuGetErrorName )(CUresult, const char **)
 
CUresult(* _cuFuncSetAttribute )(CUfunction, CUfunction_attribute, int)
 
CUresult(* _cuLinkComplete )(CUlinkState, void **, size_t *)
 
CUresult(* _cuModuleUnload )(CUmodule)
 
CUresult(* _cuLinkDestroy )(CUlinkState)
 
CUresult(* _cuLinkCreate )(unsigned int, CUjit_option *, void **, CUlinkState *)
 
CUresult(* _cuModuleLoadData )(CUmodule *, const void *)
 
CUresult(* _cuModuleGetFunction )(CUfunction *, CUmodule, const char *)
 
CUresult(* _cuLinkAddFile )(CUlinkState, CUjitInputType, const char *, unsigned int, CUjit_option *, void **)
 
CUresult(* _cuLinkAddData )(CUlinkState, CUjitInputType, void *, size_t, const char *, unsigned int, CUjit_option *, void **)
 
CUresult(* _cuLaunchCooperativeKernel )(CUfunction, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, CUstream, void **)
 
CUresult(* _cuLaunchKernel )(CUfunction f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, unsigned int blockDimZ, unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, void **extra)
 

Constructor & Destructor Documentation

◆ CUDADriverWrapper()

CUDADriverWrapper::CUDADriverWrapper ( )

◆ ~CUDADriverWrapper()

CUDADriverWrapper::~CUDADriverWrapper ( )

Member Function Documentation

◆ cuGetErrorName()

CUresult CUDADriverWrapper::cuGetErrorName ( CUresult  error,
const char **  pStr 
) const
Here is the caller graph for this function:

◆ cuFuncSetAttribute()

CUresult CUDADriverWrapper::cuFuncSetAttribute ( CUfunction  hfunc,
CUfunction_attribute  attrib,
int  value 
) const
Here is the caller graph for this function:

◆ cuLinkComplete()

CUresult CUDADriverWrapper::cuLinkComplete ( CUlinkState  state,
void **  cubinOut,
size_t *  sizeOut 
) const

◆ cuModuleUnload()

CUresult CUDADriverWrapper::cuModuleUnload ( CUmodule  hmod) const

◆ cuLinkDestroy()

CUresult CUDADriverWrapper::cuLinkDestroy ( CUlinkState  state) const

◆ cuModuleLoadData()

CUresult CUDADriverWrapper::cuModuleLoadData ( CUmodule *  module,
const void *  image 
) const
Here is the caller graph for this function:

◆ cuLinkCreate()

CUresult CUDADriverWrapper::cuLinkCreate ( unsigned int  numOptions,
CUjit_option *  options,
void **  optionValues,
CUlinkState *  stateOut 
) const

◆ cuModuleGetFunction()

CUresult CUDADriverWrapper::cuModuleGetFunction ( CUfunction *  hfunc,
CUmodule  hmod,
const char *  name 
) const
Here is the caller graph for this function:

◆ cuLinkAddFile()

CUresult CUDADriverWrapper::cuLinkAddFile ( CUlinkState  state,
CUjitInputType  type,
const char *  path,
unsigned int  numOptions,
CUjit_option *  options,
void **  optionValues 
) const

◆ cuLinkAddData()

CUresult CUDADriverWrapper::cuLinkAddData ( CUlinkState  state,
CUjitInputType  type,
void *  data,
size_t  size,
const char *  name,
unsigned int  numOptions,
CUjit_option *  options,
void **  optionValues 
) const

◆ cuLaunchCooperativeKernel()

CUresult CUDADriverWrapper::cuLaunchCooperativeKernel ( CUfunction  f,
unsigned int  gridDimX,
unsigned int  gridDimY,
unsigned int  gridDimZ,
unsigned int  blockDimX,
unsigned int  blockDimY,
unsigned int  blockDimZ,
unsigned int  sharedMemBytes,
CUstream  hStream,
void **  kernelParams 
) const

◆ cuLaunchKernel()

CUresult CUDADriverWrapper::cuLaunchKernel ( CUfunction  f,
unsigned int  gridDimX,
unsigned int  gridDimY,
unsigned int  gridDimZ,
unsigned int  blockDimX,
unsigned int  blockDimY,
unsigned int  blockDimZ,
unsigned int  sharedMemBytes,
CUstream  hStream,
void **  kernelParams,
void **  extra 
) const
Here is the caller graph for this function:

Member Data Documentation

◆ handle

void* nvinfer1::CUDADriverWrapper::handle
private

◆ _cuGetErrorName

CUresult(* nvinfer1::CUDADriverWrapper::_cuGetErrorName) (CUresult, const char **)
private

◆ _cuFuncSetAttribute

CUresult(* nvinfer1::CUDADriverWrapper::_cuFuncSetAttribute) (CUfunction, CUfunction_attribute, int)
private

◆ _cuLinkComplete

CUresult(* nvinfer1::CUDADriverWrapper::_cuLinkComplete) (CUlinkState, void **, size_t *)
private

◆ _cuModuleUnload

CUresult(* nvinfer1::CUDADriverWrapper::_cuModuleUnload) (CUmodule)
private

◆ _cuLinkDestroy

CUresult(* nvinfer1::CUDADriverWrapper::_cuLinkDestroy) (CUlinkState)
private

◆ _cuLinkCreate

CUresult(* nvinfer1::CUDADriverWrapper::_cuLinkCreate) (unsigned int, CUjit_option *, void **, CUlinkState *)
private

◆ _cuModuleLoadData

CUresult(* nvinfer1::CUDADriverWrapper::_cuModuleLoadData) (CUmodule *, const void *)
private

◆ _cuModuleGetFunction

CUresult(* nvinfer1::CUDADriverWrapper::_cuModuleGetFunction) (CUfunction *, CUmodule, const char *)
private

◆ _cuLinkAddFile

CUresult(* nvinfer1::CUDADriverWrapper::_cuLinkAddFile) (CUlinkState, CUjitInputType, const char *, unsigned int, CUjit_option *, void **)
private

◆ _cuLinkAddData

CUresult(* nvinfer1::CUDADriverWrapper::_cuLinkAddData) (CUlinkState, CUjitInputType, void *, size_t, const char *, unsigned int, CUjit_option *, void **)
private

◆ _cuLaunchCooperativeKernel

CUresult(* nvinfer1::CUDADriverWrapper::_cuLaunchCooperativeKernel) (CUfunction, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, CUstream, void **)
private

◆ _cuLaunchKernel

CUresult(* nvinfer1::CUDADriverWrapper::_cuLaunchKernel) (CUfunction f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, unsigned int blockDimZ, unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, void **extra)
private
The documentation for this class was generated from the following files: