From b7565222beee8bfaec05048c05c5b17a71de289a Mon Sep 17 00:00:00 2001
From: Bo Zhang <105368690+zhangbopd@users.noreply.github.com>
Date: Tue, 25 Apr 2023 11:04:05 +0800
Subject: [PATCH] add syncthreads (#53149)

---
 paddle/phi/kernels/funcs/dropout_impl.cu.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/phi/kernels/funcs/dropout_impl.cu.h b/paddle/phi/kernels/funcs/dropout_impl.cu.h
index 0b47febb0d3..006784eb64c 100644
--- a/paddle/phi/kernels/funcs/dropout_impl.cu.h
+++ b/paddle/phi/kernels/funcs/dropout_impl.cu.h
@@ -272,6 +272,7 @@ __global__ void DropOutNdForwardKernel(
       idx = fast_divmoder.val[0];
       offset += broadcast_config.strides[j] * fast_divmoder.val[1];
     }
+    __syncthreads();
     y[i] = dst_functor(src[i], mask[offset]);
   }
 }
-- 
GitLab