diff --git a/paddle/fluid/operators/bce_loss_op_npu.cc b/paddle/fluid/operators/bce_loss_op_npu.cc
index ee4aebbea01d8e0951f40f334610a58312fe9f1a..f6b0f7b3fbbd53a6e29cb12d6378b975281acbe6 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>);