fix gpups-pscore build bug;test=develop (#42485)

35fadcdf · danleifeng · GitHub · 6ec89eeb · 35fadcdf
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/framework/fleet/ps_gpu_wrapper.cc paddle/fluid/framework/fleet/ps_gpu_wrapper.cc +1 -1

未找到文件。
--- a/paddle/fluid/framework/fleet/ps_gpu_wrapper.cc
+++ b/paddle/fluid/framework/fleet/ps_gpu_wrapper.cc
@@ -630,7 +630,7 @@ void PSGPUWrapper::BuildPull(std::shared_ptr<HeterContext> gpu_task) {
 #endif

 #ifdef PADDLE_WITH_PSCORE
-    auto& task_ptrs = device_task_ptrs[dev];
+    auto& task_ptrs = device_task_ptrs[shard_id];
 #endif

    int len = prefix_sum[dev][shard_id + 1] - prefix_sum[dev][shard_id];