triton.testing.Benchmark

class triton.testing.Benchmark(self, x_names, x_vals, line_arg, line_vals, line_names, plot_name, args, xlabel='', ylabel='', x_log=False, y_log=False)

This class is used by the perf_report function to generate line plots with a concise API.

__init__(self, x_names, x_vals, line_arg, line_vals, line_names, plot_name, args, xlabel='', ylabel='', x_log=False, y_log=False)

Constructor

Parameters
  • x_names (List[str]) – Name of the arguments that should appear on the x axis of the plot. If the list contains more than one element, all the arguments are assumed to have the same value.

  • x_vals (List[Any]) – List of values to use for the arguments in x_names.

  • line_arg (str) – Argument name for which different values correspond to different lines in the plot.

  • line_vals (List[str]) – List of values to use for the arguments in line_arg.

  • line_names (List[str]) – Label names for the different lines.

  • plot_name (str) – Name of the plot.

  • args (List[str]) – List of arguments to remain fixed throughout the benchmark.

  • xlabel (str, optional) – Label for the x axis of the plot.

  • ylabel (str, optional) – Label for the y axis of the plot.

  • x_log (bool, optional) – Whether the x axis should be log scale.

  • y_log (bool, optional) – Whether the y axis should be log scale.

Methods

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

Constructor