version up

cpu/sampler.cpp

-#include <ATen/CPUGenerator.h>
 #include <torch/extension.h>
 
 at::Tensor neighbor_sampler(at::Tensor start, at::Tensor cumdeg, size_t size,
                             float factor) {
-  at::CPUGenerator *generator = at::detail::getDefaultCPUGenerator();
 
   auto start_ptr = start.data<int64_t>();
   auto cumdeg_ptr = cumdeg.data<int64_t>();
@@ -24,7 +22,7 @@ at::Tensor neighbor_sampler(at::Tensor start, at::Tensor cumdeg, size_t size,
     std::unordered_set<int64_t> set;
     if (size_i < 0.7 * float(num_neighbors)) {
       while (set.size() < size_i) {
-        int64_t z = generator->random() % num_neighbors;
+        int64_t z = rand() % num_neighbors;
         set.insert(z + low);
       }
       std::vector<int64_t> v(set.begin(), set.end());

setup.py

@@ -29,7 +29,7 @@ if CUDA_HOME is not None:
                       ['cuda/rw.cpp', 'cuda/rw_kernel.cu']),
     ]
 
-__version__ = '1.4.3a1'
+__version__ = '1.4.3'
 url = 'https://github.com/rusty1s/pytorch_cluster'
 
 install_requires = ['scipy']

torch_cluster/__init__.py

@@ -7,7 +7,7 @@ from .radius import radius, radius_graph
 from .rw import random_walk
 from .sampler import neighbor_sampler
 
-__version__ = '1.4.3a1'
+__version__ = '1.4.3'
 
 __all__ = [
     'graclus_cluster',