KernelLaunchParams

class tensorrt.plugin.KernelLaunchParams(grid_x: int | IDimensionExpr | SymInt32 = 1, grid_y: int | IDimensionExpr | SymInt32 = 1, grid_z: int | IDimensionExpr | SymInt32 = 1, block_x: int | IDimensionExpr | SymInt32 = 1, block_y: int | IDimensionExpr | SymInt32 = 1, block_z: int | IDimensionExpr | SymInt32 = 1, shared_mem: int | IDimensionExpr | SymInt32 = 0)

Parameters:

• grid_x (Union[int, trt.IDimensionExpr, SymInt32], optional) – The grid x dimension. Defaults to 1.

• grid_y (Union[int, trt.IDimensionExpr, SymInt32], optional) – The grid y dimension. Defaults to 1.

• grid_z (Union[int, trt.IDimensionExpr, SymInt32], optional) – The grid z dimension. Defaults to 1.

• block_x (Union[int, trt.IDimensionExpr, SymInt32], optional) – The x dimension of each thread block. Defaults to 1.

• block_y (Union[int, trt.IDimensionExpr, SymInt32], optional) – The y dimension of each thread block. Defaults to 1.

• block_z (Union[int, trt.IDimensionExpr, SymInt32], optional) – The z dimension of each thread block. Defaults to 1.

• shared_mem (Union[int, trt.IDimensionExpr, SymInt32], optional) – Shared-memory per thread block in bytes. Defaults to 0.