From 1f3f98ddcf90032cff2241e7c3e3729079a79d1c Mon Sep 17 00:00:00 2001 From: Manman Ren Date: Wed, 27 Nov 2024 10:14:53 -0800 Subject: [PATCH] lint Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: --- tritonbench/kernels/triton_fused_attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tritonbench/kernels/triton_fused_attention.py b/tritonbench/kernels/triton_fused_attention.py index a7c5512a..311852d9 100644 --- a/tritonbench/kernels/triton_fused_attention.py +++ b/tritonbench/kernels/triton_fused_attention.py @@ -536,7 +536,7 @@ def _attn_fwd_inner_ws( ) for BM in [128] for BN in [128] - for mult in [1] #, 2, 4, 8, 16] + for mult in [1] for sched in schedList for enable_tma in tmaList for enable_ws in [True]