From 83452dbb4a19c6a2461e972eb2b64a2df9a466b8 Mon Sep 17 00:00:00 2001
From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
Date: Sun, 19 Jan 2025 10:56:13 +0800
Subject: [PATCH] fix file name spelling mistake and useless variable in
 minmax-text-01-lightning_attention (#2971)

---
 .../benchmark_lightning_attention_decode.py}                    | 1 -
 .../benchmark_lightning_attention_prefill.py}                   | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)
 rename benchmark/kernels/{minmax-text-01-lighting_attention/benchmark_lighting_attention_decode.py => minmax-text-01-lightning_attention/benchmark_lightning_attention_decode.py} (99%)
 rename benchmark/kernels/{minmax-text-01-lighting_attention/benchmark_lighting_attention_prefill.py => minmax-text-01-lightning_attention/benchmark_lightning_attention_prefill.py} (99%)

diff --git a/benchmark/kernels/minmax-text-01-lighting_attention/benchmark_lighting_attention_decode.py b/benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_decode.py
similarity index 99%
rename from benchmark/kernels/minmax-text-01-lighting_attention/benchmark_lighting_attention_decode.py
rename to benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_decode.py
index 4ce7f2b49..a2d1e10f6 100644
--- a/benchmark/kernels/minmax-text-01-lighting_attention/benchmark_lighting_attention_decode.py
+++ b/benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_decode.py
@@ -26,7 +26,6 @@ def _decode_kernel(
     d_original: tl.constexpr,
     e: tl.constexpr,
     e_original: tl.constexpr,
-    BLOCK_SIZE: tl.constexpr = 32,
 ):
     off_bh = tl.program_id(0)
     off_h = off_bh % h
diff --git a/benchmark/kernels/minmax-text-01-lighting_attention/benchmark_lighting_attention_prefill.py b/benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_prefill.py
similarity index 99%
rename from benchmark/kernels/minmax-text-01-lighting_attention/benchmark_lighting_attention_prefill.py
rename to benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_prefill.py
index 3db4694c7..cd298487b 100644
--- a/benchmark/kernels/minmax-text-01-lighting_attention/benchmark_lighting_attention_prefill.py
+++ b/benchmark/kernels/minmax-text-01-lightning_attention/benchmark_lightning_attention_prefill.py
@@ -493,6 +493,8 @@ def test_lightning_attention_implementations(model_params):
         msg="Lightning attention implementations produce different results",
     )
 
+    print("✅ Two implementations match")
+
 
 def get_benchmark():
     batch_size_range = [2**i for i in range(0, 7)]  # max 64