From 9922e5fddfaaad9f3cba25e41c69ab69806a8c4c Mon Sep 17 00:00:00 2001 From: Devesh Date: Mon, 20 Nov 2023 14:08:13 -0600 Subject: [PATCH] Restore cpp change --- src/cpp/src/data/samplers/neighbor.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cpp/src/data/samplers/neighbor.cpp b/src/cpp/src/data/samplers/neighbor.cpp index 4496085b..7ed6bf5b 100644 --- a/src/cpp/src/data/samplers/neighbor.cpp +++ b/src/cpp/src/data/samplers/neighbor.cpp @@ -153,7 +153,7 @@ std::tuple sample_uniform_cpu(torch::Tensor edges, if (num_columns == 3) { #pragma omp parallel default(none) shared(tid_seeds, local_offsets_accessor, local_offsets, global_offsets_accessor, global_offsets, num_neighbors_accessor, \ - num_neighbors, max_neighbors, sorted_list_ptr, edges, ret_neighbor_id_edges_mem, ret_neighbor_id_edges) + num_neighbors, max_neighbors, sorted_list_ptr, edges, ret_neighbor_id_edges_mem, ret_neighbor_id_edges) { #ifdef MARIUS_OMP unsigned int seed = tid_seeds[omp_get_thread_num()]; @@ -193,7 +193,7 @@ std::tuple sample_uniform_cpu(torch::Tensor edges, } } else { #pragma omp parallel default(none) shared(tid_seeds, local_offsets_accessor, local_offsets, global_offsets_accessor, global_offsets, num_neighbors_accessor, \ - num_neighbors, max_neighbors, sorted_list_ptr, edges, ret_neighbor_id_edges_mem, ret_neighbor_id_edges) + num_neighbors, max_neighbors, sorted_list_ptr, edges, ret_neighbor_id_edges_mem, ret_neighbor_id_edges) { #ifdef MARIUS_OMP unsigned int seed = tid_seeds[omp_get_thread_num()];