Skip to content

Commit

Permalink
adding new configs for servicelab
Browse files Browse the repository at this point in the history
Summary:
adding new configs for servicelab + logging to scuba

Follow up diff coming up to add aggregates into logging (ie harmonic mean)

Reviewed By: xuzhao9

Differential Revision: D64126688

fbshipit-source-id: 0c3705e82071f1399cfc53ff496d130adf237b73
  • Loading branch information
Adam Mainz authored and facebook-github-bot committed Oct 9, 2024
1 parent 4fd7c74 commit 533d258
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion userbenchmark/triton/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def _run(args: argparse.Namespace, extra_args: List[str]) -> BenchmarkOperatorRe
else:
print(metrics)
if not hasattr(torch_version, "git_version") and args.log_scuba:
from pytorch.benchmark.fb.run_utils import log_benchmark
from .fb.utils import log_benchmark

log_benchmark(metrics, args.op)
if args.plot:
Expand Down

0 comments on commit 533d258

Please sign in to comment.