From 46dd1d4aadedf77c3eaec2eb5eba04faabd448d2 Mon Sep 17 00:00:00 2001
From: Qi Li <qili93@qq.com>
Date: Tue, 23 Mar 2021 12:21:30 +0800
Subject: [PATCH] [ROCM] fix reduce_sum nan in ROCM platform, test=develop
 (#31780)

---
 paddle/fluid/operators/reduce_ops/cub_reduce.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddle/fluid/operators/reduce_ops/cub_reduce.h b/paddle/fluid/operators/reduce_ops/cub_reduce.h
index 39cce60faf..29e46e091d 100644
--- a/paddle/fluid/operators/reduce_ops/cub_reduce.h
+++ b/paddle/fluid/operators/reduce_ops/cub_reduce.h
@@ -161,7 +161,11 @@ static inline std::vector<int> GetStrides(const std::vector<int>& dims,
   return strides;
 }
 
+#ifdef __HIPCC__
+constexpr int kMaxBlockDim = 256;
+#else
 constexpr int kMaxBlockDim = 512;
+#endif
 
 static inline int GetDesiredBlockDim(int block_dim) {
   return block_dim >= kMaxBlockDim
-- 
GitLab