triton.language.static_range

class triton.language.static_range(self, arg1, arg2=None, step=None)

一个永远向上计数的迭代器。

@triton.jit
def kernel(...):
    for i in tl.static_range(10):
        ...

Note:

这是一种特殊的迭代器，用于在triton.jit函数中实现类似于Python的range语义。此外，它还会引导编译器积极展开循环。

Parameters:

• arg1 – 起始值。

• arg2 – 结束值。

• step – 步长值。

__init__(self, arg1, arg2=None, step=None)

方法

__init__(self, arg1[, arg2, step])