From 9ae8f751146fdddb87665833e4a90d4618f3071e Mon Sep 17 00:00:00 2001
From: Iwan Kawrakow <iwan.kawrakow@gmail.com>
Date: Mon, 19 May 2025 15:30:46 +0300
Subject: [PATCH] Fix bf16

---
 ggml/src/iqk/fa/iqk_fa_templates.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ggml/src/iqk/fa/iqk_fa_templates.h b/ggml/src/iqk/fa/iqk_fa_templates.h
index a8b19401..3cbdddd6 100644
--- a/ggml/src/iqk/fa/iqk_fa_templates.h
+++ b/ggml/src/iqk/fa/iqk_fa_templates.h
@@ -1653,8 +1653,8 @@ struct FlashAttn {
 #ifdef __AVX512BF16__
 
 template <int D, int step>
-struct HelperBF16 final : public BaseHelper<step> {
-    using Base = BaseHelper<step>;
+struct HelperBF16 final : public BaseHelper {
+    using Base = BaseHelper;
     HelperBF16(const char * data, int stride) : Base(data, stride) {}
     inline void load(int l1, __m512bh * vk) const {
         auto dr = Base::lblock(l1);