【NPU】fix bug of using temp vector (#31963)

4 years ago · bc7a3afa68
parent a6343afc70
commit bc7a3afa68
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/operators/fill_constant_op_npu.cc
+++ b/paddle/fluid/operators/fill_constant_op_npu.cc
@ -65,7 +65,8 @@ class FillConstantNPUKernel : public framework::OpKernel<T> {

    Tensor tensor_tmp(data_type);
    tensor_tmp.mutable_data<T>({1}, ctx.GetPlace());
-    TensorFromVector(std::vector<T>{value}, ctx.device_context(), &tensor_tmp);
+    std::vector<T> init = {value};
+    TensorFromVector(init, ctx.device_context(), &tensor_tmp);

    out_var->mutable_data<T>(shape, place);
    auto runner = NpuOpRunner("FillD", {tensor_tmp}, {*out_var},