Add usage logger to track user usage

Summary: We would like to gather some data about how many users are using Tritonbench and the command line options.

Reviewed By: int3

Differential Revision: D59408795

fbshipit-source-id: 43c60936330aef521162fbe1eaf669cf33a8ffa5

userbenchmark/triton/run.py

@@ -2,6 +2,7 @@
 import os
 import sys
 import tempfile
+import torch
 from typing import List
 from torch import version as torch_version
 from torchbenchmark.operators import load_opbench_by_name
@@ -11,6 +12,11 @@
     DEFAULT_RUN_ITERS,
     DEFAULT_WARMUP,
 )
+if not hasattr(torch.version, "git_version"):
+    from pytorch.benchmark.fb.run_utils import usage_report_logger
+else:
+    usage_report_logger = lambda *args, **kwargs: None
+
 
 TRITON_BENCH_CSV_DUMP_PATH = tempfile.gettempdir() + "/tritonbench/"
 
@@ -84,6 +90,8 @@ def _run(args: argparse.Namespace, extra_args: List[str]) -> None:
 def run(args: List[str] = []):
     if args == []:
         args = sys.argv[1:]
+    # Log the tool usage
+    usage_report_logger(benchmark_name="tritonbench")
     args, extra_args = parse_args(args)
     if args.ci:
         from .ci import run_ci