From 99ed2c1ae326a68cec5597bb9ecea11aaaabe80b Mon Sep 17 00:00:00 2001
From: danyao12 <danyao12>
Date: Fri, 26 Jul 2024 18:41:18 +0800
Subject: [PATCH] code sync up

---
 .../ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
index 3c85c64189..e551d9ec79 100644
--- a/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
+++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
@@ -82,7 +82,7 @@ struct BlockFmhaPipelineQRKSVSAsync
         else
         {
             // minimize occupancy
-            if constexpr(BiasEnum != BlockAttentionBiasEnum::NO_BIAS && kHasDropout)
+            if constexpr(BiasEnum != BlockAttentionBiasEnum::NO_BIAS && FmhaDropout::IsDropout)
             {
                 return 1;
             }
@@ -303,7 +303,7 @@ struct BlockFmhaPipelineQRKSVSAsync
         constexpr auto k_pre_np = [&]() {
             if constexpr(kPadSeqLenK &&
                          (BiasEnum == BlockAttentionBiasEnum::ELEMENTWISE_BIAS ||
-                          (BiasEnum != BlockAttentionBiasEnum::NO_BIAS && kHasDropout)))
+                          (BiasEnum != BlockAttentionBiasEnum::NO_BIAS && FmhaDropout::IsDropout)))
                 return bool_constant<true>{};
             else
                 return bool_constant<false>{};