fix compile

PaddlePaddle · Feb 15, 2022 · ef6a4d3 · ef6a4d3 · paddle-bot-old · Feb 15, 2022
1 parent 8c29718
commit ef6a4d3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/paddle/fluid/platform/device/gpu/cuda/cuda_device_function.h b/paddle/fluid/platform/device/gpu/cuda/cuda_device_function.h
@@ -19,6 +19,7 @@ limitations under the License. */
 #include "paddle/fluid/platform/bfloat16.h"
 #include "paddle/fluid/platform/complex.h"
 #include "paddle/fluid/platform/float16.h"
+#include "paddle/pten/core/enforce.h"
 
 namespace paddle {
 namespace platform {
@@ -70,8 +71,8 @@ __forceinline__ __device__ bfloat16 CudaShuffleDownSync(unsigned mask,
   return bfloat16(__shfl_down_sync(mask, static_cast<nv_bfloat16>(val),
                                    static_cast<unsigned>(delta), width));
 #else
-  PADDLE_THROW(platform::errors::Unimplemented(
-      "__shfl_down_sync with bfloat16 is not supported on cuda <= 11"));
+  PADDLE_ENFORCE(
+      false, "__shfl_down_sync with bfloat16 is not supported on cuda <= 11.");
 #endif
 }