[GPUGraph] fix FillOneStep args (PaddlePaddle#107)

* fix ins no * fix FillOnestep args
danleifeng · Sep 9, 2022 · 582d236 · 582d236
1 parent b8b0d26
commit 582d236
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/framework/data_feed.cu b/paddle/fluid/framework/data_feed.cu
@@ -1059,8 +1059,8 @@ int GraphDataGenerator::FillWalkBuf() {
                    (uint64_t)sample_keys_ptr,
                    1,
                    sample_res.total_sample_size);
+      int sample_key_len =  sample_res.total_sample_size;
       sample_res = gpu_graph_ptr->graph_neighbor_sample_v3(q, false);
-
       if (FLAGS_gpugraph_storage_mode != GpuGraphStorageMode::WHOLE_HBM) {
         // table_->insert(sample_res.actual_val, sample_res.total_sample_size,
         // d_uniq_node_num, sample_stream_);
@@ -1072,7 +1072,7 @@ int GraphDataGenerator::FillWalkBuf() {
       }
       FillOneStep(d_type_keys + start,
                   cur_walk,
-                  sample_res.total_sample_size,
+                  sample_key_len,
                   sample_res,
                   1,
                   step,