From 74db66905c63fc53ae675ec7c854d6798a6c8ea1 Mon Sep 17 00:00:00 2001
From: Qi Li <qili93@qq.com>
Date: Thu, 15 Jul 2021 14:25:22 +0800
Subject: [PATCH] [NPU] fix bce loss op, test=develop (#34170)

---
 paddle/fluid/operators/bce_loss_op_npu.cc | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/paddle/fluid/operators/bce_loss_op_npu.cc b/paddle/fluid/operators/bce_loss_op_npu.cc
index ee4aebbea01..f6b0f7b3fbb 100644
--- a/paddle/fluid/operators/bce_loss_op_npu.cc
+++ b/paddle/fluid/operators/bce_loss_op_npu.cc
@@ -69,9 +69,9 @@ namespace ops = paddle::operators;
 namespace plat = paddle::platform;
 
 REGISTER_OP_NPU_KERNEL(
-    bce_loss, ops::BCELossNPUKernel<plat::CUDADeviceContext, float>,
-    ops::BCELossNPUKernel<plat::CUDADeviceContext, plat::float16>);
+    bce_loss, ops::BCELossNPUKernel<plat::NPUDeviceContext, float>,
+    ops::BCELossNPUKernel<plat::NPUDeviceContext, plat::float16>);
 
 REGISTER_OP_NPU_KERNEL(
-    bce_loss_grad, ops::BCELossGradNPUKernel<plat::CUDADeviceContext, float>,
-    ops::BCELossGradNPUKernel<plat::CUDADeviceContext, plat::float16>);
+    bce_loss_grad, ops::BCELossGradNPUKernel<plat::NPUDeviceContext, float>,
+    ops::BCELossGradNPUKernel<plat::NPUDeviceContext, plat::float16>);
-- 
GitLab