tvm_tirx.backend.cuda.launch.KernelAttributes

class tvm_tirx.backend.cuda.launch.KernelAttributes(*, min_blocks_per_sm=None, max_blocks_per_cluster=None, max_registers_per_thread=None, required_block_size=False)

Bases: Config

Compile-time CUDA kernel attributes. See the generated launch reference.

Parameters:

• min_blocks_per_sm (Any)

• max_blocks_per_cluster (Any)

• max_registers_per_thread (Any)

• required_block_size (Any)

min_blocks_per_sm: Any = None

max_blocks_per_cluster: Any = None

max_registers_per_thread: Any = None

required_block_size: Any = False