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

zhiqiu · Apr 15, 2021 · 3489ae0 · 3489ae0
1 parent f5c50b5
commit 3489ae0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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},