[ROCm] fix bfloat16 support, test=develop

PaddlePaddle · Mar 10, 2022 · b5b95fe · b5b95fe · paddle-bot-old · Mar 10, 2022
1 parent 061044a
commit b5b95fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/imperative/gradient_accumulator.cc b/paddle/fluid/imperative/gradient_accumulator.cc
@@ -423,7 +423,7 @@ void TensorAdd(const VarType& src, VarType* dst) {
   }
   if (data_type == framework::proto::VarType::BF16) {
     if (platform::is_gpu_place(place)) {
-#if defined(PADDLE_WITH_CUDA)
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
       return TensorAddImpl<platform::CUDADeviceContext, platform::bfloat16>(
           src_tensor, dst_tensor, place);
 #else