test (nn/sdpa): bugfix

Xilinx · Nov 29, 2024 · 198a648 · 198a648
1 parent 3a422f2
commit 198a648
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/nn/quant_sdpa.py b/src/brevitas/nn/quant_sdpa.py
@@ -100,7 +100,7 @@ def forward(
         kwargs = {}
         if scale is not None:
             kwargs["scale"] = scale
-        if not enable_gqa:
+        if enable_gqa:
             kwargs["enable_gqa"] = enable_gqa
         return F.scaled_dot_product_attention(
             query=query,