From c3d17437849351ac77cfff6ad7d196005ebb863b Mon Sep 17 00:00:00 2001 From: Dewei Wang Date: Sun, 17 Nov 2024 18:39:29 -0800 Subject: [PATCH] fix format --- .../triton_kernels_benchmark/flash_attention_fwd_benchmark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py b/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py index 73668bfb2e..5e2a47b2bc 100644 --- a/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py +++ b/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py @@ -214,7 +214,7 @@ def forward(q, k, v, causal, sm_scale): num_warps=num_warps, # num_stages=num_stages, # grf_mode='large', # - advanced_path=True, + advanced_path=True, # ) return o