diff --git a/paddle/fluid/operators/activation_op_npu.cc b/paddle/fluid/operators/activation_op_npu.cc index 33d8214a757b007b3d2570756a81a18db8229fdc..24ebc00c4841c9364959b112ab3490884efe88e4 100644 --- a/paddle/fluid/operators/activation_op_npu.cc +++ b/paddle/fluid/operators/activation_op_npu.cc @@ -12,7 +12,6 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the Licnse. */ -#ifdef PADDLE_WITH_ASCEND_CL #include #include @@ -90,8 +89,8 @@ class PowGradNPUKernel : public framework::OpKernel { // factor. Tensor factor_bc_tensor(framework::proto::VarType::FP32); factor_bc_tensor.mutable_data(x_dims, place); - auto runner_bc = NpuOpRunner("BroadcastTo", {factor_tensor, x_shape}, - {factor_bc_tensor}, {}); + auto runner_bc = NpuOpRunner("FillD", {factor_tensor}, {factor_bc_tensor}, + {{"dims", x_dims}}); runner_bc.Run(stream); // Step 3: Compute x_power_mul_factor = factor * x.pow(factor-1) @@ -123,5 +122,3 @@ REGISTER_OP_NPU_KERNEL( pow_grad, ops::PowGradNPUKernel, ops::PowGradNPUKernel); - -#endif