运行时 API

运行时

GluonJITFunction

jit

用于使用 Triton 编译器进行 JIT(即时)编译函数的装饰器。

constexpr_function

封装任意 Python 函数,使其能够在 Triton 函数中对 constexpr 参数进行编译时调用,并返回一个 constexpr 结果。

must_use_result

如果该函数的结果未被使用,则抛出错误。

主机端描述符

nvidia.hopper.TensorDescriptor

nvidia.hopper.TensorDescriptorIm2Col

nvidia.blackwell.TensorDescriptor