From cbeca8f004ba5c47051f21718eba50b39cb02d35 Mon Sep 17 00:00:00 2001 From: Aleksandr Eremin Date: Thu, 24 Oct 2024 17:20:36 +0300 Subject: [PATCH] fix build --- src/solver/mha/mha_ck_fa_v2_solver_forward.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/solver/mha/mha_ck_fa_v2_solver_forward.cpp b/src/solver/mha/mha_ck_fa_v2_solver_forward.cpp index 6313f3920f..0de88b56bb 100644 --- a/src/solver/mha/mha_ck_fa_v2_solver_forward.cpp +++ b/src/solver/mha/mha_ck_fa_v2_solver_forward.cpp @@ -239,8 +239,12 @@ MhaCKFlashAttentionV2Forward::GetSolution([[maybe_unused]] const ExecutionContex // and isn't async. fmha_runtime_args.p_drop = probability; - fmha_runtime_args.drop_seed_offset = - std::make_pair(dataFwd.dropoutSeedData, dataFwd.dropoutOffsetData); + // fmha_runtime_args.drop_seed_offset = + // std::make_pair(dataFwd.dropoutSeedData, dataFwd.dropoutOffsetData); + + // using dataFwd.dropoutSeedData gpu pointer was causing compiler error + // since dropout is disabled for now, placing 0. + fmha_runtime_args.drop_seed_offset = std::make_pair(0, 0); // Create stream_config, and set it to not time kernel. ck_tile::stream_config stream_config;