From c70758ca61257419f211a635d51069bf893e8bd6 Mon Sep 17 00:00:00 2001
From: Qianfeng <qianfeng.zhang@amd.com>
Date: Thu, 20 Jun 2024 11:36:42 +0800
Subject: [PATCH] Fix in dropout lambda to avoid the compiling issue on some
 docker/compiler envs (#1350)

[ROCm/composable_kernel commit: e3f44659cf77df8c3de15eb14baffd58be6ac550]
---
 include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp
index 6db432c83b..5ecc3a4d80 100644
--- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp
+++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp
@@ -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,