triton.testing.Benchmark

class triton.testing.Benchmark(self, x_names: List[str], x_vals: List[Any], line_arg: str, line_vals: List[Any], line_names: List[str], plot_name: str, args: Dict[str, Any], xlabel: str = '', ylabel: str = '', x_log: bool = False, y_log: bool = False, styles=None)

这个类被perf_report函数用来通过简洁的API生成折线图。

__init__(self, x_names: List[str], x_vals: List[Any], line_arg: str, line_vals: List[Any], line_names: List[str], plot_name: str, args: Dict[str, Any], xlabel: str = '', ylabel: str = '', x_log: bool = False, y_log: bool = False, styles=None)

构造函数。 x_vals可以是标量列表或元组/列表的列表。如果x_vals是标量列表且存在多个x_names，则所有参数将具有相同的值。 如果x_vals是元组/列表的列表，则每个元素的长度应与x_names相同。

Parameters:

• x_names (List[str]) - 应该出现在图表x轴上的参数名称。

• x_vals (List[Any]) - 用于x_names中参数的数值列表。

• line_arg (str) - 参数名称，不同的值对应图中不同的线条。

• line_vals (List[Any]) - 用于line_arg中参数的数值列表。

• line_names (List[str]) – 不同线条的标签名称。

• plot_name (str) - 图表名称。

• args (Dict[str, Any]) – 在整个基准测试过程中保持固定的关键字参数字典。

• xlabel (str, optional) – 图表x轴的标签。

• ylabel (str, optional) – 图表y轴的标签。

• x_log (bool, optional) – 是否将x轴设置为对数刻度。

• y_log (bool, optional) - 是否将y轴设置为对数刻度。

• styles (list[tuple[str, str]]) - 一个元组列表，其中每个元组包含两个元素：颜色和线条样式。

方法

__init__(self, x_names, x_vals, line_arg, ...)

构造函数。