From 582d2361e7d947d813c19cd0f7e6c6f8168467c9 Mon Sep 17 00:00:00 2001
From: Thunderbrook <52529258+Thunderbrook@users.noreply.github.com>
Date: Fri, 9 Sep 2022 11:24:12 +0800
Subject: [PATCH] [GPUGraph] fix FillOneStep args (#107)

* fix ins no

* fix FillOnestep args
---
 paddle/fluid/framework/data_feed.cu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/framework/data_feed.cu b/paddle/fluid/framework/data_feed.cu
index 59b540834dcf7..5e7be45c8f619 100644
--- 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,