fix bug in slot_feature (PaddlePaddle#71)

Co-authored-by: root <root@yq01-inf-hic-k8s-a100-ab2-0009.yq01.baidu.com>
seemingwang · Jul 26, 2022 · e7ca266 · e7ca266
1 parent 3c79186
commit e7ca266
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/framework/fleet/heter_ps/graph_gpu_ps_table_inl.cu b/paddle/fluid/framework/fleet/heter_ps/graph_gpu_ps_table_inl.cu
@@ -73,9 +73,9 @@ __global__ void get_features_kernel(GpuPsCommGraphFea graph, GpuPsFeaInfo* fea_i
                                 int* actual_size, uint64_t* feature, int slot_num, int n) {
   int idx = blockIdx.x * blockDim.y + threadIdx.y;
   if (idx < n) {
-    int node_offset = fea_info_array[idx].feature_offset;
+    int feature_size = fea_info_array[idx].feature_size;
     int offset = idx * slot_num;
-    if (node_offset == 0) {
+    if (feature_size == 0) {
       for (int k = 0; k < slot_num; ++ k) {
         feature[offset + k] = 0;
       }