use memcpy

dmlc · May 20, 2024 · 5060968 · 5060968
1 parent b6be7a1
commit 5060968
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/graphbolt/src/fused_csc_sampling_graph.cc b/graphbolt/src/fused_csc_sampling_graph.cc
@@ -1348,9 +1348,7 @@ static torch::Tensor NonUniformPickOp(
               auto j = RandomEngine::ThreadLocal()->RandInt(i, num_neighbors);
               std::swap(positive_probs_indices_ptr[i], positive_probs_indices_ptr[j]);
             }
-            for (int64_t i = 0; i < fanout; ++i) {
-              ret_ptr[i] = positive_probs_indices_ptr[i];
-            }
+            std::memcpy(ret_ptr, positive_probs_indices_ptr, fanout * sizeof(int64_t));
           } else if (fanout < 64) {
             auto begin = ret_ptr;
             auto end = ret_ptr + fanout;