fix bug (PaddlePaddle#55)

Co-authored-by: liaoxiaochao <liaoxiaochao@baidu.com>
zmxdream · Sep 27, 2022 · 0b4f09b · 0b4f09b
1 parent 93f4ced
commit 0b4f09b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/framework/fleet/ps_gpu_wrapper.cc b/paddle/fluid/framework/fleet/ps_gpu_wrapper.cc
@@ -707,7 +707,7 @@ void PSGPUWrapper::BuildPull(std::shared_ptr<HeterContext> gpu_task) {
     }
 
     auto fill_func = [] (std::mutex* mutex, std::vector<FeatureKey>& keys, 
-                          std::vector<paddle::ps::DownpourFixedFeatureValue*> values, 
+                          std::vector<paddle::ps::DownpourFixedFeatureValue*>& values, 
                           std::vector<std::pair<FeatureKey, paddle::ps::DownpourFixedFeatureValue*>>& src_kv,
                           uint32_t keys_sum) -> void {
       mutex->lock();