Commit 6ea55be6 authored by rusty1s's avatar rusty1s
Browse files

scope

parent 5ecb0823
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
at::Tensor neighbor_sampler(at::Tensor start, at::Tensor cumdeg, size_t size, at::Tensor neighbor_sampler(at::Tensor start, at::Tensor cumdeg, size_t size,
float factor) { float factor) {
CPUGenerator *generator = at::detail::getDefaultCPUGenerator(); at::CPUGenerator *generator = at::detail::getDefaultCPUGenerator();
auto start_ptr = start.data<int64_t>(); auto start_ptr = start.data<int64_t>();
auto cumdeg_ptr = cumdeg.data<int64_t>(); auto cumdeg_ptr = cumdeg.data<int64_t>();
......
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