triton.Config

class triton.Config(self, meta, num_warps=4, num_stages=2)

An object that represents a possible kernel configuration for the auto-tuner to try.

Variables
  • meta – a dictionary of meta-parameters to pass to the kernel as keyword arguments.

  • num_warps – the number of warps to use for the kernel when compiled for GPUs. For example, if num_warps=8, then each kernel instance will be automatically parallelized to cooperatively execute using 8 * 32 = 256 threads.

  • num_stages – the number of stages that the compiler should use when software-pipelining loops. Mostly useful for matrix multiplication workloads on SM80+ GPUs.

__init__(self, meta, num_warps=4, num_stages=2)

Methods

__init__(self, meta[, num_warps, num_stages])