Skip to content

GitLab

Unverified Commit 7b470eb0 authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

[GraphBolt] Quick fix for GCC 8 (#7279)

parent e847fc44
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
graphbolt/src/cuda/extension/unique_and_compact_map.cu
View file @ 7b470eb0
......@@ -166,15 +166,16 @@ UniqueAndCompactBatchedHashMapBased(
pointers_and_offsets.data_ptr<int64_t>() + 3 * num_batches;
for (std::size_t i = 0; i < num_batches; i++) {
pointers_ptr[2 * i] = unique_dst_ids.at(i).data_ptr<index_t>();
offsets_ptr[2 * i] = unique_dst_ids[i].size(0);
offsets_ptr[2 * i + 1] = unique_dst_ids[i].size(0);
pointers_ptr[2 * i + 1] = src_ids.at(i).data_ptr<index_t>();
offsets_ptr[2 * i + 1] = src_ids[i].size(0);
offsets_ptr[2 * i + 2] = src_ids[i].size(0);
pointers_ptr[2 * num_batches + i] = dst_ids.at(i).data_ptr<index_t>();
offsets_ptr[2 * num_batches + i] = dst_ids[i].size(0);
offsets_ptr[2 * num_batches + i + 1] = dst_ids[i].size(0);
}
// Finish computing the offsets by taking a cumulative sum.
std::exclusive_scan(
offsets_ptr, offsets_ptr + 3 * num_batches + 1, offsets_ptr, 0ll);
offsets_ptr[0] = 0;
std::partial_sum(
offsets_ptr, offsets_ptr + 3 * num_batches + 1, offsets_ptr);
// Device version of the tensors defined above. We store the information
// initially on the CPU, which are later copied to the device.
auto pointers_and_offsets_dev = torch::empty(
......
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