Reuse Benchmark for benchmarks from triton.testing (#3162)

Part of #3160

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

benchmarks/triton_kernels_benchmark/benchmark_testing.py

@@ -1,7 +1,8 @@
 import argparse
 import itertools
 import os
-from typing import Any, Dict, List
+
+from triton.testing import Benchmark
 
 BENCHMARKING_METHOD = os.getenv("BENCHMARKING_METHOD", "UPSTREAM_PYTORCH_PROFILER")
 
@@ -171,73 +172,6 @@ def perf_report(benchmarks):
     return wrapper
 
 
-# # pylint: disable=too-many-instance-attributes
-class Benchmark:
-    """
-    This class is used by the :code:`perf_report` function to generate line plots with a concise API.
-    """
-
-    def __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,
-        color=None,  # pylint: disable=unused-argument
-        styles=None,
-    ):
-        """
-        Constructor.
-        x_vals can be a list of scalars or a list of tuples/lists. If x_vals is a list
-        of scalars and there are multiple x_names, all arguments will have the same value.
-        If x_vals is a list of tuples/lists, each element should have the same length as
-        x_names.
-
-        :param x_names: Name of the arguments that should appear on the x axis of the plot.
-        :type x_names: List[str]
-        :param x_vals: List of values to use for the arguments in :code:`x_names`.
-        :type x_vals: List[Any]
-        :param line_arg: Argument name for which different values correspond to different lines in the plot.
-        :type line_arg: str
-        :param line_vals: List of values to use for the arguments in :code:`line_arg`.
-        :type line_vals: List[Any]
-        :param line_names: Label names for the different lines.
-        :type line_names: List[str]
-        :param plot_name: Name of the plot.
-        :type plot_name: str
-        :param args: Dictionary of keyword arguments to remain fixed throughout the benchmark.
-        :type args: Dict[str, Any]
-        :param xlabel: Label for the x axis of the plot.
-        :type xlabel: str, optional
-        :param ylabel: Label for the y axis of the plot.
-        :type ylabel: str, optional
-        :param x_log: Whether the x axis should be log scale.
-        :type x_log: bool, optional
-        :param y_log: Whether the y axis should be log scale.
-        :type y_log: bool, optional
-        """
-        self.x_names = x_names
-        self.x_vals = x_vals
-        self.x_log = x_log
-        self.line_arg = line_arg
-        self.line_vals = line_vals
-        self.line_names = line_names
-        self.y_log = y_log
-        self.styles = styles
-        # plot info
-        self.xlabel = xlabel
-        self.ylabel = ylabel
-        self.plot_name = plot_name
-        self.args = args
-
-
 class Mark:
 
     def __init__(self, fn, benchmarks):