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 ...@@ -744,11 +744,11 @@ struct FmhaFwdKernel
} }
}(); }();
auto dropout = [&]() { auto dropout = [&, i_nhead_ = i_nhead, i_batch_ = i_batch]() {
if constexpr(kHasDropout) if constexpr(kHasDropout)
{ {
return BlockDropout{i_batch, return BlockDropout{i_batch_,
i_nhead, i_nhead_,
kargs.num_head_q, kargs.num_head_q,
kargs.drop_seed, kargs.drop_seed,
kargs.drop_offset, 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