fix format

Dewei-Wang-sh · Dewei-Wang-sh · commit c3d174378493 · 2024-11-17T18:39:29.000-08:00
diff --git 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
 

Original file line number	Diff line number	Diff line change
`@@ -214,7 +214,7 @@ def forward(q, k, v, causal, sm_scale):`
`214`	`214`	`num_warps=num_warps, #`
`215`	`215`	`num_stages=num_stages, #`
`216`	`216`	`grf_mode='large', #`
`217`		`- advanced_path=True,`
	`217`	`+ advanced_path=True, #`
`218`	`218`	`)`
`219`	`219`	`return o`
`220`	`220`