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{}; else return bool_constant{};