triton.Config

class triton.Config(self, kwargs, num_warps=4, num_stages=3, num_ctas=1, maxnreg=None, pre_hook=None, ir_override=None)

一个表示可供自动调优器尝试的内核配置的对象。

变量：

• kwargs – 一个作为关键字参数传递给内核的元参数字典。

• num_warps – 为 GPU 编译内核时使用的 warp 数量。例如，如果 num_warps=8，那么每个内核实例将自动并行化，通过协作方式使用 8 * 32 = 256 线程执行。

• num_stages – 编译器在软件流水线化循环时应使用的阶段数。主要用于 SM80+ GPU 上的矩阵乘法工作负载。

• num_ctas – 块簇中的块数量。仅限 SM90+。

• maxnreg – 单个线程可使用的最大寄存器数量。对应于 ptx .maxnreg 指令。并非所有平台都支持。

• pre_hook – 在内核调用前将被调用的函数。此函数的参数为 args。

• ir_override – 用户定义 IR 的文件名 (*.{ttgir|llir|ptx|amdgcn})。

__init__(self, kwargs, num_warps=4, num_stages=3, num_ctas=1, maxnreg=None, pre_hook=None, ir_override=None)

方法

__init__(self, kwargs[, num_warps, ...])
all_kwargs(self)