diff --git a/paddle/phi/kernels/weighted_sample_neighbors_kernel.h b/paddle/phi/kernels/weighted_sample_neighbors_kernel.h
index 2a0402f9fc4947..468d4b2a00968b 100644
--- a/paddle/phi/kernels/weighted_sample_neighbors_kernel.h
+++ b/paddle/phi/kernels/weighted_sample_neighbors_kernel.h
@@ -32,4 +32,16 @@ void GraphWeightedSampleNeighborsKernel(
     DenseTensor* out_count,
     DenseTensor* out_eids);
 
+template <typename T, typename Context>
+void WeightedSampleNeighborsKernel(const Context& dev_ctx,
+                                   const DenseTensor& row,
+                                   const DenseTensor& col_ptr,
+                                   const DenseTensor& edge_weight,
+                                   const DenseTensor& x,
+                                   const paddle::optional<DenseTensor>& eids,
+                                   int sample_size,
+                                   bool return_eids,
+                                   DenseTensor* out,
+                                   DenseTensor* out_count,
+                                   DenseTensor* out_eids);
 }  // namespace phi