Skip to content

GitLab

Unverified Commit 4682b76e authored by Gan Quan's avatar Gan Quan Committed by GitHub
Browse files

fixing stupid segfault in edge_ids (#216)

parent 168794cd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/graph/graph.cc
View file @ 4682b76e
...@@ -189,6 +189,8 @@ Graph::EdgeArray Graph::EdgeIds(IdArray src_ids, IdArray dst_ids) const { ...@@ -189,6 +189,8 @@ Graph::EdgeArray Graph::EdgeIds(IdArray src_ids, IdArray dst_ids) const {
for (i = 0, j = 0; i < srclen && j < dstlen; i += src_stride, j += dst_stride) { for (i = 0, j = 0; i < srclen && j < dstlen; i += src_stride, j += dst_stride) {
const dgl_id_t src_id = src_data[i], dst_id = dst_data[j]; const dgl_id_t src_id = src_data[i], dst_id = dst_data[j];
CHECK(HasVertex(src_id) && HasVertex(dst_id)) <<
"invalid edge: " << src_id << " -> " << dst_id;
const auto& succ = adjlist_[src_id].succ; const auto& succ = adjlist_[src_id].succ;
for (size_t k = 0; k < succ.size(); ++k) { for (size_t k = 0; k < succ.size(); ++k) {
if (succ[k] == dst_id) { if (succ[k] == dst_id) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment