Unverified Commit e3f44659 authored by Qianfeng's avatar Qianfeng Committed by GitHub
Browse files

Fix in dropout lambda to avoid the compiling issue on some docker/compiler envs (#1350)

parent 8db331a5
......@@ -744,11 +744,11 @@ struct FmhaFwdKernel
}
}();
auto dropout = [&]() {
auto dropout = [&, i_nhead_ = i_nhead, i_batch_ = i_batch]() {
if constexpr(kHasDropout)
{
return BlockDropout{i_batch,
i_nhead,
return BlockDropout{i_batch_,
i_nhead_,
kargs.num_head_q,
kargs.drop_seed,
kargs.drop_offset,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment