From a41fe81d2b19eaabe2cf2ba7eb41585c24a6dce2 Mon Sep 17 00:00:00 2001
From: Loser Cheems <losercheems@gmail.com>
Date: Mon, 23 Jun 2025 13:56:04 +0800
Subject: [PATCH] Removes unused template parameter from DynamicMask

Simplifies the template signature by removing the kNThreads parameter that was not being used within the struct definition, reducing template complexity without affecting functionality.
---
 csrc/src/mask.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/csrc/src/mask.h b/csrc/src/mask.h
index c90d69b..d2394cf 100644
--- a/csrc/src/mask.h
+++ b/csrc/src/mask.h
@@ -20,7 +20,7 @@ namespace FLASH_NAMESPACE {
 
 using namespace cute;
 
-template <bool Is_causal, int kNThreads>
+template <bool Is_causal>
 struct DynamicMask {
     const int max_seqlen_k, max_seqlen_q;
     const int keep_window_size;