diff --git a/paddle/fluid/operators/bce_loss_op_npu.cc b/paddle/fluid/operators/bce_loss_op_npu.cc index f6b0f7b3fbbd53a6e29cb12d6378b975281acbe6..3136c02af41e60b09c57b903879f28354b3abca9 100644 --- a/paddle/fluid/operators/bce_loss_op_npu.cc +++ b/paddle/fluid/operators/bce_loss_op_npu.cc @@ -34,8 +34,9 @@ class BCELossNPUKernel : public framework::OpKernel { ctx.template device_context() .stream(); - const auto& runner = NpuOpRunner("BinaryCrossEntropy", {*x, *labels}, - {*out}, {{"reduction", "none"}}); + const auto& runner = + NpuOpRunner("BinaryCrossEntropy", {*x, *labels}, {*out}, + {{"reduction", static_cast("none")}}); runner.Run(stream); } }; @@ -57,7 +58,7 @@ class BCELossGradNPUKernel : public framework::OpKernel { const auto& runner = NpuOpRunner("BinaryCrossEntropyGrad", {*x, *labels, *dout}, {*dx}, - {{"reduction", "none"}}); + {{"reduction", static_cast("none")}}); runner.Run(stream); } }; diff --git a/python/paddle/fluid/tests/unittests/npu/test_bce_loss_npu.py b/python/paddle/fluid/tests/unittests/npu/test_bce_loss_npu.py index 16db952533437c699fb7993350f280a03ccbe76a..7c3d32647aea9d7411492152a083fb2fed610cb6 100644 --- a/python/paddle/fluid/tests/unittests/npu/test_bce_loss_npu.py +++ b/python/paddle/fluid/tests/unittests/npu/test_bce_loss_npu.py @@ -96,7 +96,7 @@ def test_dygraph_layer(place, label_np, reduction='mean', weight_np=None): - paddle.disable_static() + paddle.disable_static(place) if weight_np is not None: weight = paddle.to_tensor(weight_np) bce_loss = paddle.nn.loss.BCELoss(weight=weight, reduction=reduction) @@ -113,7 +113,7 @@ def test_dygraph_functional(place, label_np, reduction='mean', weight_np=None): - paddle.disable_static() + paddle.disable_static(place) input = paddle.to_tensor(input_np) label = paddle.to_tensor(label_np)