From 0ac84f6c5a4dffd3a4361b7683495321c56dda2a Mon Sep 17 00:00:00 2001 From: Tom Fogal <60981+tfogal@users.noreply.github.com> Date: Tue, 4 Feb 2025 16:56:56 -0800 Subject: [PATCH] missed a spot w/ earlier lintrunner. --- benchmarks/python/test_cat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/benchmarks/python/test_cat.py b/benchmarks/python/test_cat.py index e625f7fdb2f..ebdc5a80cfd 100644 --- a/benchmarks/python/test_cat.py +++ b/benchmarks/python/test_cat.py @@ -402,7 +402,9 @@ def benchmark_fn(inputs): @pytest.mark.parametrize("executor", ["thunder", "torchcompile"]) -def test_cat_phi3_1_baseline_benchmark(benchmark, executor: str, disable_validation: bool) -> None: +def test_cat_phi3_1_baseline_benchmark( + benchmark, executor: str, disable_validation: bool +) -> None: def to_be_compiled(t14): # t14: "cuda:0 f32[1, 48, 2048]" t0 = torch.permute(t14, (0, 2, 1)) # t0: "cuda:0 f32[1, 2048, 48]"