From 7fb9e5b7e7206edd9ac2592bb2793cb0d675188c Mon Sep 17 00:00:00 2001 From: Adam Mainz Date: Thu, 5 Dec 2024 19:20:25 -0800 Subject: [PATCH] starting to add autotune configs to logger Summary: TSIA adding configs to tritonbench logger not adding the autotune configs in directly yet but this sets it all up nicely Reviewed By: nmacchioni Differential Revision: D66856128 fbshipit-source-id: 3edfb881959d3467999f4ca48d23eecfce98a8f8 --- run.py | 1 + 1 file changed, 1 insertion(+) diff --git a/run.py b/run.py index 2399d717..3231cc76 100644 --- a/run.py +++ b/run.py @@ -75,6 +75,7 @@ def _run(args: argparse.Namespace, extra_args: List[str]) -> BenchmarkOperatorRe "benchmark_name": args.op, "device": args.device, "logging_group": args.logging_group, + "precision": args.precision, } if args.production_shapes: from tritonbench.utils.fb.durin_data import productionDataLoader