From 6962d3e23dcd2b2b3784895806f902a3a6b98d90 Mon Sep 17 00:00:00 2001 From: pangengzheng <117730991+pangengzheng@users.noreply.github.com> Date: Mon, 26 Jun 2023 18:49:26 +0800 Subject: [PATCH] exclude xpu (#54848) --- paddle/phi/kernels/kps/elementwise_add_kernel.cu | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddle/phi/kernels/kps/elementwise_add_kernel.cu b/paddle/phi/kernels/kps/elementwise_add_kernel.cu index 31d1e4e32cf..b3fe46a1cd3 100644 --- a/paddle/phi/kernels/kps/elementwise_add_kernel.cu +++ b/paddle/phi/kernels/kps/elementwise_add_kernel.cu @@ -72,14 +72,18 @@ void AddKernel(const Context& dev_ctx, const DenseTensor& x, const DenseTensor& y, DenseTensor* out) { +#ifdef PADDLE_WITH_CUDA if (x.dtype() == phi::DataType::FLOAT32 && (y.dtype() == phi::DataType::BFLOAT16 || y.dtype() == phi::DataType::FLOAT16)) { using Type = DataTypeToCppType::type; Float32Bfloat16OrFloat16AddCudaFunctor(dev_ctx, x, y, out); } else { +#endif AddCudaFunctor(dev_ctx, x, y, -1, out); +#ifdef PADDLE_WITH_CUDA } +#endif } template -- GitLab