From 166ef78b236d94e365198c5285f0dd223813e9c6 Mon Sep 17 00:00:00 2001 From: LeiWang1999 Date: Mon, 24 Feb 2025 09:40:42 +0000 Subject: [PATCH] lint fix --- benchmark/blocksparse_attention/benchmark_library_dense_fmha.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmark/blocksparse_attention/benchmark_library_dense_fmha.py b/benchmark/blocksparse_attention/benchmark_library_dense_fmha.py index f3759a5..f22a3a9 100644 --- a/benchmark/blocksparse_attention/benchmark_library_dense_fmha.py +++ b/benchmark/blocksparse_attention/benchmark_library_dense_fmha.py @@ -4,6 +4,7 @@ import torch from tilelang.profiler import do_bench + def get_sparse_attn_mask_from_topk(x, topk, use_dense_for_last_block=False): bsz, num_head, downsample_len, _ = x.shape # N_CTX = downsample_len * BLOCK