nvinfer1::IKernelLaunchParams Class Reference

Allows for kernel launch parameters to be communicated to the TensorRT backend. More...

#include <NvInferPythonPlugin.h>

Public Member Functions
ISymExpr *	getGridX () noexcept
	Get the X dimension of the grid. More...
bool	setGridX (ISymExpr *gridX) noexcept
	Set the X dimension of the grid. More...
ISymExpr *	getGridY () noexcept
	Get the Y dimension of the grid. More...
bool	setGridY (ISymExpr *gridY) noexcept
	Set the Y dimension of the grid. More...
ISymExpr *	getGridZ () noexcept
	Get the Z dimension of the grid. More...
bool	setGridZ (ISymExpr *gridZ) noexcept
	Set the Z dimension of the grid. More...
ISymExpr *	getBlockX () noexcept
	Get the X dimension of each thread block. More...
bool	setBlockX (ISymExpr *blockX) noexcept
	Set the X dimension of each thread block. More...
ISymExpr *	getBlockY () noexcept
	Get the Y dimension of each thread block. More...
bool	setBlockY (ISymExpr *blockY) noexcept
	Set the Y dimension of each thread block. More...
ISymExpr *	getBlockZ () noexcept
	Get the Z dimension of each thread block. More...
bool	setBlockZ (ISymExpr *blockZ) noexcept
	Set the Z dimension of each thread block. More...
ISymExpr *	getSharedMem () noexcept
	Get the dynamic shared-memory per thread block in bytes. More...
bool	setSharedMem (ISymExpr *sharedMem) noexcept
	Set the dynamic shared-memory per thread block in bytes. More...

Protected Member Functions
virtual	~IKernelLaunchParams () noexcept=0

Protected Attributes
IKernelLaunchParamsImpl *	mImpl {nullptr}

Detailed Description

Allows for kernel launch parameters to be communicated to the TensorRT backend.

Note

Clients must not implement this class.

Constructor & Destructor Documentation

~IKernelLaunchParams()

nvinfer1::IKernelLaunchParams::~IKernelLaunchParams ( )

inlineprotectedpure virtualdefaultnoexcept

Member Function Documentation

getBlockX()

ISymExpr * nvinfer1::IKernelLaunchParams::getBlockX ( )

inlinenoexcept

Get the X dimension of each thread block.

getBlockY()

ISymExpr * nvinfer1::IKernelLaunchParams::getBlockY ( )

inlinenoexcept

Get the Y dimension of each thread block.

getBlockZ()

ISymExpr * nvinfer1::IKernelLaunchParams::getBlockZ ( )

inlinenoexcept

Get the Z dimension of each thread block.

getGridX()

ISymExpr * nvinfer1::IKernelLaunchParams::getGridX ( )

inlinenoexcept

Get the X dimension of the grid.

getGridY()

ISymExpr * nvinfer1::IKernelLaunchParams::getGridY ( )

inlinenoexcept

Get the Y dimension of the grid.

getGridZ()

ISymExpr * nvinfer1::IKernelLaunchParams::getGridZ ( )

inlinenoexcept

Get the Z dimension of the grid.

getSharedMem()

ISymExpr * nvinfer1::IKernelLaunchParams::getSharedMem ( )

inlinenoexcept

Get the dynamic shared-memory per thread block in bytes.

setBlockX()

bool nvinfer1::IKernelLaunchParams::setBlockX ( ISymExpr *  blockX )

inlinenoexcept

Set the X dimension of each thread block.

Returns

true if each thread block's X dimension was set successfully, false otherwise

setBlockY()

bool nvinfer1::IKernelLaunchParams::setBlockY ( ISymExpr *  blockY )

inlinenoexcept

Set the Y dimension of each thread block.

Returns

true if each thread block's Y dimension was set successfully, false otherwise

setBlockZ()

bool nvinfer1::IKernelLaunchParams::setBlockZ ( ISymExpr *  blockZ )

inlinenoexcept

Set the Z dimension of each thread block.

Returns

true if each thread block's Z dimension was set successfully, false otherwise

setGridX()

bool nvinfer1::IKernelLaunchParams::setGridX ( ISymExpr *  gridX )

inlinenoexcept

Set the X dimension of the grid.

Returns

true if the grid's X dimension was set successfully, false otherwise

setGridY()

bool nvinfer1::IKernelLaunchParams::setGridY ( ISymExpr *  gridY )

inlinenoexcept

Set the Y dimension of the grid.

Returns

true if the grid's Y dimension was set successfully, false otherwise

setGridZ()

bool nvinfer1::IKernelLaunchParams::setGridZ ( ISymExpr *  gridZ )

inlinenoexcept

Set the Z dimension of the grid.

Returns

true if the grid's Z dimension was set successfully, false otherwise

setSharedMem()

bool nvinfer1::IKernelLaunchParams::setSharedMem ( ISymExpr *  sharedMem )

inlinenoexcept

Set the dynamic shared-memory per thread block in bytes.

Returns

true if the dynamic shared-memory per thread block was set successfully, false otherwise

Member Data Documentation

mImpl

IKernelLaunchParamsImpl* nvinfer1::IKernelLaunchParams::mImpl {nullptr}

protected

---

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

• NvInferPythonPlugin.h