Fix_StackOverFlow_case5

paddle/phi/kernels/cpu/graph_reindex_kernel.cc

@@ -58,6 +58,22 @@ void GraphReindexKernel(const Context& dev_ctx,
     }
     src[i] = node_map[node];
   }
+  /*
+  check if x dim and count dim all equal 1
+  the count's data must less or equal to 10
+  */
+  const int cs = count.dims()[0];
+  if ((bs == 1) && (cs == 1)) {
+    PADDLE_ENFORCE_LE(
+        count_data[0],
+        10,
+        phi::errors::OutOfRange(
+            "When X's dimension and count's dimension all equal to 1,"
+            "The count's data should be smaller or equal to 10."
+            "But received count's data = %d",
+            count_data[0]));
+  }
+
   // Reindex Dst
   // Add support for multi-type edges reindex
   int num_edge_types = count.dims()[0] / bs;

python/paddle/fluid/tests/unittests/test_graph_reindex.py

@@ -532,5 +532,21 @@ def test_heter_reindex_result_static(self):
             np.testing.assert_allclose(out_nodes, out_nodes_2, rtol=1e-05)
 
 
+class TestGraphReindex2(unittest.TestCase):
+    def setUp(self):
+        self.x = np.arange(16).astype("int32")
+
+    def test_reindex_countValue(self):
+        paddle.disable_static()
+        res = []
+        x = paddle.to_tensor(self.x)
+        y = np.array(res).astype("int32")
+
+        def test_countValue_dynamic():
+            paddle.incubate.graph_reindex(x, x, x, y, y, False)
+
+        self.assertRaises(ValueError, test_countValue_dynamic)
+
+
 if __name__ == "__main__":
     unittest.main()